백준 [11060] 점프 점프 (자바)

문제

입력

첫째 줄에 N(1 ≤ N ≤ 1,000)이 주어진다. 둘째 줄에는 Ai (0 ≤ Ai ≤ 100)가 주어진다.

출력

재환이가 최소 몇 번 점프를 해야 가장 오른쪽 끝 칸으로 갈 수 있는지 출력한다. 만약, 가장 오른쪽 끝으로 갈 수 없는 경우에는 -1을 출력한다.

풀이

1. 해당 idx의 값이 오른쪽으로 이동할 수 있는 최소한의 값이다.
2. 1부터 이동할 수 있는 값까지 더해주면서 이동할 수 있다면 큐에 삽입
3. N-1이면 점프 횟수를 출력한다.
4. N-1까지 탐색할 수 없다면 -1 출력한다.

처음 bfs를 생각하고 풀었는데 방문노드를 사용하지 않아서 실패했었다..

dp로 풀이하는 방식도 함께 추가하였다.

dp의 경우 최소 값을 구하는 경우라서 dp 배열을 모두 Integer.MAX_VALUE로 초기화 시킨 것 때문에 long 배열로 선언했다.

 

정답 코드

bfs 풀이

package com.baekjoon.p11060;

import java.io.*;
import java.util.*;

public class Main {
    static int N;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        N = Integer.parseInt(br.readLine());

        long[] dp = new long[N + 1];
        int[] arr = new int[N + 1];

        StringTokenizer st;
        st = new StringTokenizer(br.readLine());
        for (int i = 1; i <= N; i++) {
            arr[i] = Integer.parseInt(st.nextToken());
            dp[i] = Integer.MAX_VALUE;
        }

        dp[1] = 0;
        for (int i = 1; i <= N; i++) {
            for (int j = 1; j <= arr[i]; j++) {
                if (i + j <= N)
                    dp[i + j] = Math.min(dp[i + j], dp[i] + 1);
            }
        }

        if (dp[N] == Integer.MAX_VALUE) {
            System.out.println(-1);
        } else {
            System.out.println(dp[N]);
        }
    }
}

 

dp풀이

package com.baekjoon.p11060;

import java.io.*;
import java.util.*;

public class Main {
    static int N;
    static int[] arr;
    static boolean[] v;

    static class Node {
        int idx;
        int cnt;

        Node(int idx, int cnt) {
            this.idx = idx;
            this.cnt = cnt;
        }
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        N = Integer.parseInt(br.readLine());

        arr = new int[N];

        StringTokenizer st = new StringTokenizer(br.readLine());
        for (int i = 0; i < N; i++) {
            arr[i] = Integer.parseInt(st.nextToken());
        }

        bfs(0);
    }

    static void bfs(int start) {
        Queue<Node> q = new ArrayDeque<>();
        v = new boolean[N];

        v[start] = true;
        q.add(new Node(start, 0));

        while (!q.isEmpty()) {
            Node now = q.poll();

            if (now.idx == N - 1) {
                System.out.println(now.cnt);
                return;
            }

            for (int i = 1; i <= arr[now.idx]; i++) {
                int nextIdx = now.idx + i;

                if (nextIdx < N && !v[nextIdx]) {
                    v[nextIdx] = true;
                    q.add(new Node(nextIdx, now.cnt + 1));
                }
            }
        }

        System.out.println(-1);
    }
}