백준 13549번: 숨바꼭질3 (JAVA) TODO

문제

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

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

입력

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

출력

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

 

 

---

 

풀이

풀이 1. BFS에서 텔레포트 부분 먼저 처리하고, 그 다음 이동 부분 처리하는 방법.

풀이 2. 0-1 BFS 알고리즘이라는 게 있다고 한다. TODO

 

배운 점

teleport < 50001 루프 도는 부분에서 teleport == 0일때는 무한 루프가 도는데, 이를 40분동안 발견하지 못하고 고민한 듯하다. while문의 탈출 조건을 잘 생각해주자.

 

코드

import java.util.ArrayDeque;
import java.util.Queue;
import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int N = sc.nextInt();
		int K = sc.nextInt();
		int[] location = new int[100001]; // 100000 이상으로 순간이동해서 내려오는 것보다 내려간 후 순간이동하는 것이 더 빠름.
		boolean[] teleported = new boolean[100001];
		
		Queue<Integer> queue = new ArrayDeque<Integer>();
		queue.offer(N);
		location[N] = 1;
		while (!queue.isEmpty()) {
			// 순간이동을 미리 처리
			int size = queue.size();
			while (size-- > 0) {
				int temp = queue.poll();
				int teleport = temp;
				queue.offer(temp);
				if (teleported[temp]) continue;
				while (teleport < 50001) { // 100000 넘어가면 의미 없음
					System.out.println(teleport);
					teleport *= 2;
					teleported[teleport] = true;
					if (location[teleport] >= 1) break;
					location[teleport] = location[temp];
					queue.offer(teleport);
				}
				teleported[temp] = true;
			}
			
			// queue에 넣어서 처리
			size = queue.size();
			while (size-- > 0) {
				int temp = queue.poll();
				if (temp-1 >= 0 && location[temp-1] < 1) {
					location[temp-1] = location[temp] + 1;
					queue.offer(temp-1);
				}
				if (temp+1 < 100001 && location[temp+1] < 1) {
					location[temp+1] = location[temp] + 1;
					queue.offer(temp+1);
				}
			}
			if (location[K] != 0) break;
		}
		System.out.println(location[K]-1); // 정의상 1 빼줌
	}

}