문제/백준

[백준/BOJ] 2775번: 부녀회장이 될테야 (파이썬3/Python3)

개 살구 2023. 1. 1. 17:23

문제

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

 

2775번: 부녀회장이 될테야

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

www.acmicpc.net


생각

test_case 수가 2라면 층 호 층 호 처럼 입력이 들어온다

그래서 for문을 활용하여 층(floor) 호(unit)를 입력 받는다

0층의 리스트를 호만큼 만든다

0층의 i호에는 i명이 산다

k층의 n호 사람수를 찾기 위해 계산하고 가장 마지막 값을 출력하면 끝


코드

test_case = int(input())

for _ in range(test_case):
  floor = int(input())
  unit = int(input())
  floor_0 = [i for i in range(1, unit + 1)]
  for k in range(floor):  # 층 수 만큼 반복
    for i in range(1, unit):
      floor_0[i] += floor_0[i - 1]
  print(floor_0[-1])  # k층에 n호 사람 수