JiSoo's Devlog

[백준 / 파이썬] 2775번 부녀회장이 될테야 본문

코테준비

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

지숭숭숭 2024. 1. 13. 20:07

t = int(input())

for i in range(t):
    k = int(input())
    n = int(input())
    p = [i for i in range(1, n+1)] # 0층
    for j in range(k):
        for c in range(1, n):
            p[c] += p[c-1]
    print(p[-1])

 

호수별로 늘어나는 규칙성이 없어 0층부터 구할 층수까지 사람 수 더해가는 방식

0층에서 1호부터 입력받은 호수까지 사람 수 리스트 생성

입력받은 층수만큼 for문을 반복하고 중첩 for문으로 1호부터 입력받은 호수까지 반복

층수가 증가할 때마다 한층 아래의 이전 호실에 사는 사람의 수를 더해준다

반복문이 끝났을 때 가장 마지막 수가 해당하는 층, 호수에 사는 사람의 수

728x90