[BaekJoon] 13913번 - 숨바꼭질 4 [Java][C++]

문제 링크

---

1. 아이디어

BaekJoon 1697번 - 숨바꼭질 문제에서 최단 시간과 최단 경로까지 구해야 하는 문제로 최단 경로를 구하기 위해 이전에 어떤 위치에서 왔는지 저장하는 prv 배열을 활용했다.

---

2. 코드

1. 풀이 [Java]

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

public class Main {

    static final int MX = 100_000;

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

        int n = Integer.parseInt(st.nextToken());
        int k = Integer.parseInt(st.nextToken());

        System.out.println(bfs(n, k));
    }

    static String bfs(int n, int k) {
        Queue<Integer> q = new ArrayDeque<>();
        q.offer(n);

        boolean[] vis = new boolean[1 + MX];
        vis[n] = true;

        int dist = 0;

        int[] prv = new int[1 + MX];
        Arrays.fill(prv, -1);

        while (!q.isEmpty()) {
            int sz = q.size();

            while (sz-- > 0) {
                int cur = q.poll();
                if (cur == k) return traceback(dist, k, prv);

                int nxt1 = cur - 1;
                if (nxt1 >= 0 && !vis[nxt1]) {
                    q.offer(nxt1);
                    vis[nxt1] = true;
                    prv[nxt1] = cur;
                }

                int nxt2 = cur + 1;
                if (nxt2 <= MX && !vis[nxt2]) {
                    q.offer(nxt2);
                    vis[nxt2] = true;
                    prv[nxt2] = cur;
                }

                int nxt3 = cur * 2;
                if (nxt3 <= MX && !vis[nxt3]) {
                    q.offer(nxt3);
                    vis[nxt3] = true;
                    prv[nxt3] = cur;
                }
            }

            dist++;
        }

        return null;
    }

    static String traceback(int dist, int k, int[] prv) {
        StringBuilder sb = new StringBuilder();
        Deque<Integer> stack = new ArrayDeque<>();

        sb.append(dist).append("\n");

        int pos = k;
        while (pos != -1) {
            stack.push(pos);
            pos = prv[pos];
        }

        while (!stack.isEmpty()) {
            sb.append(stack.pop()).append(" ");
        }

        return sb.toString();
    }
}

2. 풀이 [C++]

#include <bits/stdc++.h>
using namespace std;

const int MX = 100000;
bool vis[1 + MX];
int prv[1 + MX];

void traceback(int dist, int k) {
    vector<int> v;

    int pos = k;
    while (pos != -1) {
        v.push_back(pos);
        pos = prv[pos];
    }

    reverse(v.begin(), v.end());

    cout << dist << '\n';

    for (int x : v) {
        cout << x << ' ';
    }
}

void bfs(int n, int k) {
    queue<int> q;
    q.push(n);

    vis[n] = true;

    int dist = 0;

    memset(prv, -1, sizeof(prv));

    while (!q.empty()) {
        int sz = q.size();

        while (sz-- > 0) {
            int cur = q.front();
            q.pop();

            if (cur == k) {
                traceback(dist, k);
                return;
            }

            for (int nxt : {cur - 1, cur + 1, cur * 2}) {
                if (0 <= nxt && nxt <= MX && !vis[nxt]) {
                    q.push(nxt);
                    vis[nxt] = true;
                    prv[nxt] = cur;
                }
            }
        }

        dist++;
    }
}

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);

    int n, k;
    cin >> n >> k;

    bfs(n, k);
}

---

3. 디버깅

없음.

---

4. 참고

없음.

---