[백준] - 2775 부녀회장이 될테야! (Python)

문제 링크

코드

import sys
t=int(sys.stdin.readline().strip())
for _ in range(t):
    p=[]
    k=int(sys.stdin.readline().strip())
    n=int(sys.stdin.readline().strip())
    for i in range(k+1):
        for j in range(1, n+1):
            if i==0:
                p.append(j)
            else:
                p.append(sum(p[(i-1)*n:j+(i-1)*n:]))
    print(p[-1])

sum 함수 까먹지 말기