부녀회장이될테야
Language
Python
Compiler
CPython 3.10.2
Options
부녀회장이될테야
T = int(input())
# 테스트 케이스 개수를 정수로 변환해서 받는다.
for _ in range(T): # 테스트 개수 만큼 루프
k = int(input()) # k(층수) 를 받는다.
n = int(input()) # n(호후) 를 받는다.
if k == 0: # 0 층일 경우 호수가 인원수이므로 인원수 출력 후 끝냄
print(n)
continue # 찾았으니, 다시 앞으로 보내서 추가 루프
# 층수가 0보다 크면 계산
ms = [i for i in range(n + 1)] # 각호의 인원을 저장할 리스트, 1부터 시작하니 +1까지
# 초기값은 0층의 호수인원(호수와 같다)으로 초기화.
end = False # 테스트 케이스 한개의 루프 끝났는지 확인
for i in range(1, k + 1): # 1층 부터, 해당층 까지 반복 계산
for j in range(1, n + 1): # 1호 부터 끝호까지 더해줌
ms[j] += ms[j - 1] # 이전 호수 인원 + 아래층 호수 인원 = 현재 호수이원이 된다.
if i == k and j == n: # 찾는 층, 호수이면 출력후 끝낸다.
print(ms[j]) # 찾은 인원수 출력
end = True
break # 찾았으니 현재 루프 종료
if end == True:
break # 찾았으니 현재 루프 종료
2
1
3
2
3
$ python3 prog.py 6
10
Exit Code:
0