[기본수학1] 부녀회장이 될테야

2021. 2. 24. 01:01백준

 

2775번: 부녀회장이 될테야

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

www.acmicpc.net

제목이 굉장히 귀여운 문제였다.

 

이렇게 사람이 많이 살고있는 아파트의 부녀회장이 되려면 쉽지 않을 것 같다.

 

 

예제를 바탕으로 몇 번 그려보면 규칙성을 쉽게 발견할 수 있었다.

 

재귀적으로 풀어봤다.

 

제출코드

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))