백준 2775 파이썬 / 부녀회장이 될테야

코드

text_num = int(input())

for _ in range(text_num):
  k = int(input()) # 층 변수
  n = int(input()) # 호 변수
  f0 = [i for i in range(1,n+1)] # 0층 1호부터 ~ n호까지 인원수 만들기
  
  for i in range(k):
    for j in range(1, n):
      # fo[0] (1호)은 계속 1이며, 아래 층에 인원 수를 구하는 k층까지 더함.
      f0[j] += f0[j-1] 
  print(f0[n-1])

문제 풀이

f0 변수에 0층 1호부터 ~ n호까지의 인원수를 만들어준다.
k층의 n호에 있는 인원수를 구하기 위해선 k번 만큼 반복해줘야한다. f0[0]은 1호이기에 계속 1이며, 이를 제외하고 f0[j] += fo[j-1]을 통해 각 층의 호에 있어야 할 인원을 구한다.