BOJ 13913 숨바꼭질 4

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

문제

수빈이는 동생과 숨바꼭질을 하고 있다. 수빈이는 현재 점 N(0 ≤ N ≤ 100,000)에 있고, 동생은 점 K(0 ≤ K ≤ 100,000)에 있다. 수빈이는 걷거나 순간이동을 할 수 있다. 만약, 수빈이의 위치가 X일 때 걷는다면 1초 후에 X-1 또는 X+1로 이동하게 된다. 순간이동을 하는 경우에는 1초 후에 2*X의 위치로 이동하게 된다.

수빈이와 동생의 위치가 주어졌을 때, 수빈이가 동생을 찾을 수 있는 가장 빠른 시간이 몇 초 후인지 구하는 프로그램을 작성하시오.

---

입력

첫 번째 줄에 수빈이가 있는 위치 N과 동생이 있는 위치 K가 주어진다. N과 K는 정수이다.

---

출력

첫째 줄에 수빈이가 동생을 찾는 가장 빠른 시간을 출력한다.

둘째 줄에 어떻게 이동해야 하는지 공백으로 구분해 출력한다.

---

예제 입력 1

5 17

예제 출력 1

4
5 4 8 16 17

---

코드

이전 문제인 BOJ 12851 숨바꼭질2 보다 오히려 쉬워진 느낌이다.

이전 문제는 BFS로 같은 레벨에 있는 모든 정답들을 찾아야 한다면, 이번 문제는 하나만 찾으면 되기 때문이다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        final int MAX = 100_000;
        int N = Integer.parseInt(st.nextToken()), K = Integer.parseInt(st.nextToken()), level = 0;
        if(N == K) { // 이미 K의 위치에 있을 경우
            System.out.println("0\n" + N);
            return;
        }
        /**
         * 방문을 확인하기 위한 배열이다.
         * index는 방문한 위치이고, visited[index]는 해당 위치에 방문하기 전 index이다.
         * 기본적으로 -1로 visited 배열을 채워주고, 맨처음 방문한 경우의 visited[index]는 Integer.MIN_VALUE로 설정한다.
         * */
        int[] visited = new int[MAX + 1];
        Arrays.fill(visited, -1);
        visited[N] = Integer.MIN_VALUE;
        Queue<Integer> queue = new LinkedList<>();
        queue.offer(N);

        while (!queue.isEmpty()) {
            int check = queue.poll();

            int[] newPositions = new int[3];
            newPositions[0] = check - 1;
            newPositions[1] = check + 1;
            newPositions[2] = check * 2;

            for (int newPos: newPositions) {
                if(newPos == K) { // K를 찾을 경우, visited 배열로 정답을 구한 뒤 정답 출력 후 main 메소드를 종료시킨다.
                    ArrayList<Integer> answer = new ArrayList<>();
                    answer.add(newPos);
                    int checkIdx = check;
                    while (checkIdx != Integer.MIN_VALUE) {
                        answer.add(checkIdx);
                        checkIdx = visited[checkIdx];
                        level++;
                    }
                    StringBuilder sb = new StringBuilder();
                    sb.append(level).append('\n');
                    for (int i = answer.size() - 1; i >= 0; i--) {
                        sb.append(answer.get(i)).append(' ');
                    }
                    System.out.println(sb);
                    return;
                }

                if(newPos >= 0 && newPos <= MAX && visited[newPos] == -1) {
                    queue.offer(newPos);
                    visited[newPos] = check;
                }
            }
        }
    }
}