백준|2775번|부녀회장이 될테야

문제설명
아파트의 a층의 b호에는 (a-1)층의 1호부터 b호까지 인원수의 합만큼의 사람이 살아야한다. 아파트는 0층부터 있고 0층의 i호에는 i명이 있다. 층과 호수를 입력받았을 때 그곳에 사는 사람의 수를 구하는 문제입니다.

작동 순서
1. 입력받을 층수와 호수의 개수를 입력받습니다.
2. 층수와 호수를 입력받습니다.
3. 0층에는 i호에 i명이 살고 있으므로 미리 지정해줍니다.
4. 반복문을 이용하여 1층부터 n층까지 각 호에 몇명이 사는지 구하여 배열에 저장합니다.
5. 입력받은 층수와 호수의 정보가 저장된 배열의 위치를 찾아서 값을 출력합니다.

소스코드

import sys
T = int(sys.stdin.readline())
k = []
n = []
for i in range(T):
    k.append(int(sys.stdin.readline()))
    n.append(int(sys.stdin.readline()))
head = [[i for i in range(max(n)+1)]]
for i in range(1, max(k)+1):
    head.append([])
    for j in range(max(n)+1):
        num = 0
        for l in range(j+1):
            num += head[i - 1][l]
        head[i].append(num)
for i in range(T):
    print(head[k[i]][n[i]])

후기
굉장히 간단한 문제였기에 큰 어려움 없이 풀 수 있었던것 같습니다.