[백준] 기본 수학 1 - 2775번 (Python)

https://www.acmicpc.net/problem/2775

2775번: 부녀회장이 될테야

---

문제

평소 반상회에 참석하는 것을 좋아하는 주희는 이번 기회에 부녀회장이 되고 싶어 각 층의 사람들을 불러 모아 반상회를 주최하려고 한다.

이 아파트에 거주를 하려면 조건이 있는데, “a층의 b호에 살려면 자신의 아래(a-1)층의 1호부터 b호까지 사람들의 수의 합만큼 사람들을 데려와 살아야 한다” 는 계약 조항을 꼭 지키고 들어와야 한다.

아파트에 비어있는 집은 없고 모든 거주민들이 이 계약 조건을 지키고 왔다고 가정했을 때, 주어지는 양의 정수 k와 n에 대해 k층에 n호에는 몇 명이 살고 있는지 출력하라. 단, 아파트에는 0층부터 있고 각층에는 1호부터 있으며, 0층의 i호에는 i명이 산다.

입력

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

출력

각각의 Test case에 대해서 해당 집에 거주민 수를 출력하라.

코드 정답

t = int(input())

for _ in range(t):
    floor = int(input()) # 층
    num = int(input()) # 호
    f0 = [x for x in range(1, num+1)] # 0층 리스트
    for k in range(floor): # 층 수 만큼 반복
        for i in range(1, num): # 1 ~ num-1 까지
            f0[i] += f0[i-1] # 층별 각 호실의 사람 수 변경
    print(f0[-1]) # 가장 마지막 수 출력

풀이

1행 : 문자를 입력받아 정수형으로 변환 후 t 에 저장한다.

3행 : 횟수 0 부터 t 의 값에 도달할 때까지 4~10행을 반복한다. 반복될 때마다 0 에 1씩 더해진다.

4행 : 문자를 입력받아 정수형으로 변환 후 변수 floor 에 저장한다. 층으로 쓰일 변수 값이다.

5행 : 문자를 입력받아 정수형으로 변환 후 변수 num 에 저장한다. 호로 쓰일 변수 값이다.

6행 : x 의 값에 1부터 num+1 의 이전의 값 즉, num 의 값 전부를 f0 라는 변수를 리스트 형식으로 저장한다.

7행 : k 가 0부터 floor 의 값에 도달할 때까지 8~9행을 반복한다.

8행 : i 가 1부터 num 의 값에 도달할 때까지 9행을 반복한다.

9행 : f0[i] = f0[i] + f0[i-1] 이며 i 는 1부터 이므로 f0[0] 의 값은 변하지 않고 f[1] 의 값부터 값이 변경된다.

10행 : fo[-1] 의 값. 저장된 값이 [1, 4, 10] 이라면 가장 마지막의 10을 출력한다.