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

문제

문제 풀이

t = int(input())  # Test case의 수 t를 입력받는다.

for i in range(t):  # 각각의 케이스마다 실행되게끔 반복문을 작성한다.(테스트 케이스의 수 만큼 반복된다)
    k = int(input())  # 구하고자 하는 층 수를 k에 입력받는다.
    n = int(input())  # 구하고자 하는 호수를 n에 입력받는다.

    room = [x for x in range(1, n+1)]
    # 각 호수마다 몇명의 사람이 있는지를 나타내는 리스트를 만든다.
    # 먼저 입력받은 호수만큼 공간을 할당한다.
    # 이 때 room의 각 요소 값들은 '0층 i호에는 i명만큼 거주한다.'는 규칙에 맞게 값을 넣었다.

    for j in range(k):  # 구하고자 하는 층 수만큼 반복한다.
        # 마지막 인덱스로 n-1이 오는 이유는 room의 마지막 인덱스가 n-1에 해당하기 때문이다.
        for y in range(1, n):
            room[y] += room[y-1]  # 구하고자 하는 호실의 인원은 기존 인원에다가 직전 호의 인원을 더해주면 된다.

    print(room[-1])  # room 리스트의 마지막 값을 출력한다.

좋은 웹페이지 즐겨찾기