[백준 2775] 부녀회장이 될테야 - 수학

코드

def solution(k, n):
    arr = [i for i in range(1, n+1)]
    for _ in range(k):
        for idx, num in enumerate(arr):
            if idx == 0:
                continue
            arr[idx] = arr[idx-1] + arr[idx]
    
    print(arr[n-1])


loop = int(input())
for _ in range(loop):
    k = int(input())
    n = int(input())
    solution(k, n)

 

설명

• 0층은 기본 default 이기 때문에 n의 크기만큼 초기화 하고 첫 칸부터 1, 2, 3, ... n 으로 초기화 한다.
• 1층부터 계산을 시작하는데 1번 방은 무조건 1명이다. 그래서 2번 방부터 계산을 시작한다.
• 계산할 때는 현재 계산해야 하는 방번호 -1 의 수와 현재 계산해야하는 방번호의 수를 더한다.