[골드 5] 13549번 숨바꼭질 3⭐️

문제

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

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

입력

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

출력

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

풀이

• 선택지는 3가지 : +1, -1, 현 위치 * 2
• 걸으면 1초, 순간이동하면 0초

큐를 이용해 BFS를 구현.
[이동 위치, 소요 시간]을 넣고 이동 위치가 같아지는 순간의 소요 시간을 return한다.
(걸을 때와 순간이동할 때의 소요 시간이 다르므로 주의한다.)

=> 시간초과로 틀렸다.

const input = require('fs')
  .readFileSync('ex.txt')
  .toString()
  .trim()
  .split(' ')
  .map((el) => +el);

const [N, K] = input;
const visited = Array(100100).fill(false); // 방문 여부 나타낼 배열
const choice = [-1, 1];

function solution(cur) {
  let queue = [[cur, 0]];

  while (queue.length) {
    const [cur, move] = queue.shift();
    if (visited[cur]) continue; // 이미 방문했으면 패스
    if (cur === K) return move;
    visited[cur] = true; // 방문 체크

    for (let i = 0; i < 2; i++) {
      const next = cur + choice[i];
      queue.push([next, move + 1]);
    }
    const jump = cur * 2;
    queue.push([jump, move]);
  }
}

const answer = solution(N);

console.log(answer);

시간 개선된 풀이

걸으면 1초가 소요되고, 순간이동하면 0초가 소요된다.
따라서 가장 적은 시간으로 이동하는 경우를 구하려면 시간이 덜 걸리는 순간이동을 우선적으로 고려해야 한다.
-> 우선순위 큐/덱
-> 순간이동을 우선적으로 고려해 앞에 넣고(unshift), 나머지는 뒤에 넣는다(push).

const input = require('fs').readFileSync('ex.txt').toString().trim();

// 시간초과를 막기 위해 덱 사용해야 함.
const sol = (input) => {
  const [N, K] = input.split(' ').map(Number);
  const visit = Array.from({length: 100100}, () => 0);

  function bfs(N) {
    const queue = [[N, 0]];
    visit[N] = 1;
    while (queue.length) {
      const [cur, time] = queue.shift();
      if (cur === K) return time;
      for (next of [cur * 2, cur - 1, cur + 1]) {
        if (!visit[next] && next >= 0 && next <= 100000) {
          visit[next] = 1;
          if (next == cur * 2) {
            queue.unshift([next, time]); // 2X로 이동할 때는 시간을 증가시키지 않고, 우선순위를 반영하여 큐의 맨 앞에 넣어준다.
          } else {
            queue.push([next, time + 1]); // X-1, X+1로 이동할 때는 시간을 증가시키고, 큐에 순서대로 넣어준다.
          }
        }
      }
    }
  }
  return bfs(N);
};

console.log(sol(input));

https://gobae.tistory.com/39