백준 (boj) 파이썬 - 2775 부녀회장이 될테야

문제 링크 : https://www.acmicpc.net/problem/2775

2775번: 부녀회장이 될테야

 

 

그냥 2차원 배열을 만들어서, array[층][호실] 이라는 인덱스로 구현을해서 풀었다. 다른사람의 풀이보다 메모리를 쓸데없이 많이 쓴것같긴 하지만 문제에서의 입력범위가 크지 않아 괜찮을거라고 생각했다.

t = int(input())


for i in range(t):
    k = int(input())
    n = int(input())
    arr=[[0 for i in range(n)]for i in range(k+1)]

    for q in range(n): #0층의 인원수 채우기.
        arr[0][q] = q+1 

    for j in range(1,k+1): #0층의 인원수를 더해 1층...2층.... 쌓아가는 반복문
        for z in range(n):
            arr[j][z] += sum(arr[j-1][:z+1])

    print(arr[k][n-1])