2775. 부녀회장이 될테야

이 아파트에 거주하려면 조건이 있습니다.
"a층의 b호에 거주하려면 (a-1)층의 1호부터 b호까지의 사람들 수의 합만큼 사람들을 데려와 살아야 한다"

 

아파트에 비어있는 집은 없고 모든 거주민들이 계약 조건을 지키고 있다고 가정했을 때 k층 n호에는 몇 명이 살고 있는지 출력하세요.

단, 아파트는 0층부터 있고 각층에는 1호부터 있으며 0층 i호에는 i명이 산다.

 

 

T = int(input())

for _ in range(T):
    k = int(input())
    b = int(input())
    floor_nums = [i+1 for i in range(b)]
    
    for i in range(1,k+1):
        floor_nums = [sum(floor_nums[:j+1]) for j in range(b)]
    print(floor_nums[b-1])

 

• 우선 먼저 0층의 각 호실에 사는 사람들의 수를 floor_nums 라는 변수에 저장해줍니다.
  그리고 1층부터 시작해서 k층까지 i번째 층의 j호실에 사는 사람을 i-1번째 층의 1호실~j호실까지의 합으로 계산해서 저장해줍니다.
  최종적으로 k층에 거주하는 사람들을 구했을 때 리스트 인덱스값이 0부터 시작하므로 b-1로 인덱싱해서 k층 b호에 거주중인 사람의 수를 알아냅니다.