코딩테스트

[백준][파이썬]2775번: 부녀회장이 될테야

과아아앙 2021. 4. 11. 19:45

문제 출처 : www.acmicpc.net/problem/2775

 

2775번: 부녀회장이 될테야

첫 번째 줄에 Test case의 수 T가 주어진다. 그리고 각각의 케이스마다 입력으로 첫 번째 줄에 정수 k, 두 번째 줄에 정수 n이 주어진다

www.acmicpc.net

내 풀이

직접 그려보며 확인해본 결과

k층 n호에 사는 사람의 수는 (k층 n - 1 호에 사는 사람의 수) + (k - 1층 n호에 사는 사람의 수)

라는 규칙이 나왔다.

import sys

input = sys.stdin.readline

t = int(input())

for i in range(t):
    k = int(input())
    n = int(input())
    arr = [[i for i in range(n + 1)] for _ in range(k + 1)]
    for j in range(1, k + 1):
        for l in range(1, n + 1):
            arr[j][l] = arr[j - 1][l] + arr[j][l - 1]
    print(arr[-1][-1])