Algorithm/BOJ
[백준/2775] 부녀회장이 될테야 (with 파이썬)
Potato potage
2022. 10. 25. 00:10
반응형
✔ 문제
✔ 풀이
t = int(input())
for _ in range(t):
k = int(input())
n = int(input())
people = [i for i in range(1, n+1)] # 0층의 ! ~ n호의 인원 수
for i in range(k):
for j in range(1, n):
people[j] += people[j-1]
print(people[-1])
✔ 설명
people 리스트에 0층 1~n호까지의 인원수를 저장한다.
층수(k)만큼 반복하며 한 번의 반복마다 n-1호까지 각 층 별의 인원 수를 변경한다.
people[1] (2호실) += people[0] (1호실)
인덱스랑 호실이 똑같지 않아서 헷갈릴 수 있다.
✔ 후기
+= 사용하지 않고 sum()으로 해결할 수 있지 않을까 했는데
문제를 급하게 풀어야 해서 더 고민해보지 못하고 2중 for문으로 해결했다.
반응형