[Python/파이썬] 백준 14248번 점프 점프

https://www.acmicpc.net/problem/14248

 

코드

from collections import deque

n = int(input())
a = list(map(int, input().split()))
start = int(input())-1

q = deque([start])
visited = [False]*n
visited[start] = True
ans = 1
while q:
    now = q.popleft()
    # 왼쪽, 오른쪽
    for nx in [now-a[now], now+a[now]]:
        if 0 <= nx < n and not visited[nx]:
            visited[nx] = True
            q.append(nx)
            ans += 1
print(ans)

 

단순히 각 돌에 방문할 수 있는지 여부만 판단하면 되기 때문에 BFS를 이용하여 갈 수 있는 돌들을 방문하며 개수를 세면 된다.