What I Learned/Algorithm Practice

[백준 - python] 2775번: 부녀회장이 될테야

Interrobang 2022. 9. 14. 20:05

<문제 링크>

 

2775번: 부녀회장이 될테야

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

www.acmicpc.net

 

<문제 풀이>

T = int(input())

for _ in range(T):  
    k = int(input())
    n = int(input())
    people = [x for x in range(1, n+1)]  
    for _ in range(k):
        for i in range(1, n): 
            people[i] += people[i-1] 
    print(people[n-1])

2775 입출력 예시

*key point: 반복문을 활용한다. 0층의 사는 사람의 수를 리스트로 만든 뒤 k층까지 한층씩 올라가면서 n호에 사는 사람의 수로 계속 요소의 값을 변화시켜주었다.