문제 풀이/Baekjoon Online Judge

[python3] 2775번 부녀회장이 될테야

[잉간] 2019. 4. 19. 12:10


1부터 14까지가 저장된 2차원 리스트 A를 생성한다. 그리고 k와 n의 값이 작기에 먼저 값을 채우고 출력하는 방식으로 했다.

k와 n 모두 1부터 14까지 반복되며 k번째 열에 (k-1)번째 열에서 0부터 n까지 더한 값을 append()를 사용해 추가해 나간다.

반복문이 종료되면 테스트 케이스의 수를 입력받고 입력받은 값에 저장된 값을 출력한다.

문제풀이

A = [[i for i in range(1,15)]]

for k in range(1,15):
    A.append([])
    for n in range(1,15):
        A[k].append(sum(A[k-1][0:n]))

for T in range(int(input())):
    k,n = int(input()),int(input())
    print(A[k][n-1])