728x90
반응형

0층의 i호에는 사람이 i명 산다는 말을 통해 0층의 상황은 [1, 2, 3, 4,.... n]이라는 것을 알 수 있다.

그 이후에는 자기 앞에 호에 사는 사람의 수를 k층 n호가 나올 때까지 계속 더하면서 갱신하면 된다.

EX) k = 2, n = 3

0층 -> [1, 2, 3], 1층 -> [1, 3, 6], 2층 -> [1, 4, 10] -> 10 출력

for _ in range(int(input())):
    k = int(input())
    n = int(input())
    ans = 0
    li = []
    for i in range(1, n+1):
        li.append(i)
    for i in range(k):
        for j in range(n-1):
            li[j+1] += li[j]
    print(li[n-1])

728x90
반응형

+ Recent posts