11060. 점프 점프
문제
재환이가 1×N 크기의 미로에 갇혀있다. 미로는 1×1 크기의 칸으로 이루어져 있고, 각 칸에는 정수가 하나 쓰여 있다. i번째 칸에 쓰여 있는 수를 Ai라고 했을 때, 재환이는 Ai이하만큼 오른쪽으로 떨어진 칸으로 한 번에 점프할 수 있다. 예를 들어, 3번째 칸에 쓰여 있는 수가 3이면, 재환이는 4, 5, 6번 칸 중 하나로 점프할 수 있다.
재환이는 지금 미로의 가장 왼쪽 끝에 있고, 가장 오른쪽 끝으로 가려고 한다. 이때, 최소 몇 번 점프를 해야 갈 수 있는지 구하는 프로그램을 작성하시오. 만약, 가장 오른쪽 끝으로 갈 수 없는 경우에는 -1을 출력한다.입력
첫째 줄에 N(1 ≤ N ≤ 1,000)이 주어진다. 둘째 줄에는 Ai (0 ≤ Ai ≤ 100)가 주어진다.
출력
재환이가 최소 몇 번 점프를 해야 가장 오른쪽 끝 칸으로 갈 수 있는지 출력한다. 만약, 가장 오른쪽 끝으로 갈 수 없는 경우에는 -1을 출력한다.
BFS 알고리즘을 이용해 소스코드 작성
BFS를 통해 현재 위치에서 탐색 가능한 요소를 추가하는 방식으로 탐색
(주어진 N의 위치일 때, 탐색 종료)
import sys
from collections import deque
input = sys.stdin.readline
n = int(input())
jump = list(map(int, input().split()))
visited = [True] + [False] * (n - 1)
def bfs(start):
q = deque([start])
while q:
idx, cnt = q.popleft()
if idx == n - 1:
return cnt
for i in range(1, jump[idx] + 1):
if idx + i < n and visited[idx + i] == False:
visited[idx + i] = True
q.append((idx + i, cnt + 1))
return -1
print(bfs((0, 0)))
DP 알고리즘으로 분류되어있어서,, DP만 생각하다 못 풀었다
다른 알고리즘도 다 염두하고 풀어야겠다
import sys
from collections import deque
input = sys.stdin.readline
n = int(input())
info = [0] + list(map(int, input().split()))
INF = int(1e9)
dp = [INF for _ in range(n + 1)]
q = deque([(1, 0)]) # (현재 번호, 점프 수)
dp[1] = 0
while q:
now, cnt = q.popleft()
jmp = info[now]
for i in range(jmp):
if now + i + 1 < n + 1 and dp[now + i + 1] > cnt + 1:
dp[now + i + 1] = cnt + 1
q.append((now + i + 1, cnt + 1))
print(dp[n] if dp[n] < INF else -1)
꺄항.. 3번 제출만에 풀었당
1번은 if문 조건에서 dp[now + i + 1] > cnt + 1
조건을 체크안함
2번은 오른쪽 끝을 갈 수 없는 경우를 고려하지 않았다.
ㅎㅎ