[기본수학1] 부녀회장이 될테야
2021. 2. 24. 01:01ㆍ백준
제목이 굉장히 귀여운 문제였다.
이렇게 사람이 많이 살고있는 아파트의 부녀회장이 되려면 쉽지 않을 것 같다.
예제를 바탕으로 몇 번 그려보면 규칙성을 쉽게 발견할 수 있었다.
재귀적으로 풀어봤다.
제출코드
from sys import stdin
test_count = int(stdin.readline())
test_case = []
for i in range(0, test_count):
case_k = int(stdin.readline())
case_n = int(stdin.readline())
test_case.append([case_k, case_n])
def recursive(k: int, n: int):
if n == 1:
return 1
if n == 2:
return k+n
if k == 1:
return int(n*(n + 1) / 2)
else:
return recursive(k - 1, n) + recursive(k, n - 1)
for i in range(0, len(test_case)):
test_k = test_case[i][0]
test_n = test_case[i][1]
print(recursive(test_k, test_n))