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

코드

t = int(input())

for _ in range(t):
    floor = int(input())  # 층수
    num = int(input())  # 호수
    floor_0 = [x for x in range(1, num+1)]  # 0층

    for k in range(floor):
        for i in range(1, num):
            floor_0[i] += floor_0[i-1]
        print(floor_0)
    print(floor_0[-1])​

풀이

3층	1	5	15	35	70
2층	1	4	10	20	35
1층	1	3	6	10	15
0층	1	2	3	4	5

아파트의 층/호수별 거주민 수는 위 표와 같다.

k층 n호에 사는 거주민의 수는 k-1층의 1,2,3,...,n호에 사는 거주민의 총 합과 같다.

주어진 정보는 0층에 사는 거주민 수 (1,2,3,4,,,)이니까

층 수(floor)만큼 반복문을 활용하여 새로운 리스트를 만들어준다.