[백준] 14248번 점프점프 (파이썬)

문제

14248번: 점프 점프

코드

import sys
input = sys.stdin.readline
sys.setrecursionlimit(10**6)

def DFS(start):
    global cnt
    visited[start] = True
    cnt += 1

    for _ in range(2):
        if (0 <= start + stone[start] < n and not visited[start + stone[start]]):
            DFS(start + stone[start])
        elif 0 <= start - stone[start] < n and not visited[start - stone[start]]:
            DFS(start - stone[start])


n = int(input())
stone = list(map(int, input().split()))
start = int(input())

visited = [False] * n
cnt = 0
DFS(start-1)

print(cnt)

해설

start는 1부터 시작이니까 start - 1로 DFS에 들어갔다.

True할 때 마다 cnt += 1하고 칸수에서 넘어가지 않으면서 visited가 False라면 재귀한다.

여기서 왼쪽으로 갈 때 빼게되면서 음수가 되지 않으려나? 생각해서 abs를 줬는데 그러면 표밖을 넘어가는거라 틀리니 유의하자!