백준 2775 부녀회장이 될테야

문제

---

솔루션

t = int(input())
for _ in range(t):
    k= int(input())
    n= int(input())
    li = list()
    num = [i for i in range(1,n+1)]
    for _ in range(k):
        for j in range(1,n):
            num[j]+=num[j-1]
    print(num[-1])

 

---

깨달은 점

처음에 문제를 이해하지 못했는데 

3층 -> 1 5 15 35

2층 -> 1 4 10 20

1층 -> 1 3 6 10

0층 -> 1 2 3 4

이런식으로 그 전층의 그 호수까지의 합이 사람수였다.

 

   num = [i for i in range(1,n+1)]

이부분은 처음 0층 i호에 사는 사람 수를 리스트에 담아둔 것이다.

 

 

    for _ in range(k):
        for j in range(1,n):
            num[j]+=num[j-1]​

k로 층마다 반복문을 돌리고 그 안에 이중 반복문으로 호수 사람의 합을 세는 방법이다.

 

print(num[-1])

마지막으로 num리스트의 마지막인 마지막 층의 호수 사람의 합을 출력한다.