[백준 2331] [dfs] 반복수열

💻 백준 2331번 [반복수열]

2331번: 반복수열

풀이)

첫번째 제출 : 런타임에러(ModuleNotFoundError) 뜸

알고보니 백준에서는 numpy를 못 쓰게 함...;;^^

# 첫번째
import numpy as np

a,p = map(int,input().split())
visited = {}
stack = []
def cal(a,p) :
    if a in visited :  
        return stack.pop() 
    else :
        visited[a] = 1
        stack.append(a)
        return cal(np.sum([int(i)**p for i in str(a)]),p)
dup_n = cal(a,p)
print(list(visited.keys()).index(np.sum([int(i)**p for i in str(dup_n)])))
# ==> 런타임 에러(ModuleNotFoundError) : numpy없대...;;

 두번째 제출

# 최종제출
a,p = map(int,input().split())
visited = {}
stack = []
def cal(a,p) :
    if a in visited :  
        return stack.pop() 
    else :
        visited[a] = 1
        stack.append(a)
        return cal(sum([int(i)**p for i in str(a)]),p)
dup_n = cal(a,p)
print(list(visited.keys()).index(sum([int(i)**p for i in str(dup_n)])))
# ==> 메모리 : 28776K4B, 시간 : 64ms, 코드길이 : 323B

 

dfs를 공부할 때 되도록이면 재귀함수를 사용해보려고 노력함

위 코드에서는 visited를 dictionary로 사용하였고, stack과 재귀함수 cal을 사용하였다