흙금이네 블로그

[BOJ] 1351 - 무한 수열 (Python) 본문

알고리즘

[BOJ] 1351 - 무한 수열 (Python)

흙금 2023. 3. 16. 18:53

 

 

아이디어

 

재귀를 통해 계산에 필요한 수들을 구해 나간다.

 

 

풀이

 

def find_A(i):
    if A.get(i, 0):
        return A[i]
    A[i] = find_A(i//P)+find_A(i//Q)
    return A[i]

N, P, Q = map(int, input().split())
A = dict()
A[0] = 1
print(find_A(N))

 

Comments