문제
수빈이는 동생과 숨바꼭질을 하고 있다. 수빈이는 현재 점 N(0 ≤ N ≤ 100,000)에 있고, 동생은 점 K(0 ≤ K ≤ 100,000)에 있다. 수빈이는 걷거나 순간이동을 할 수 있다. 만약, 수빈이의 위치가 X일 때 걷는다면 1초 후에 X-1 또는 X+1로 이동하게 된다. 순간이동을 하는 경우에는 1초 후에 2*X의 위치로 이동하게 된다.
수빈이와 동생의 위치가 주어졌을 때, 수빈이가 동생을 찾을 수 있는 가장 빠른 시간이 몇 초 후인지 구하는 프로그램을 작성하시오.
입력
첫 번째 줄에 수빈이가 있는 위치 N과 동생이 있는 위치 K가 주어진다. N과 K는 정수이다.
출력
수빈이가 동생을 찾는 가장 빠른 시간을 출력한다.

package com.chanhue.algorithm.백준.실버
import java.util.LinkedList
import java.util.Queue
fun main() {
val (n, k) = readLine()!!.split(" ").map { it.toInt() }
val queue: Queue<Pair<Int,Int>> = LinkedList()
val visited = BooleanArray(100001) { false }
visited[n] = true
queue.add(Pair(n,0))
while (queue.isNotEmpty()) {
var (x,number)=queue.poll()
if(x==k){
println(number)
return
}
number++
if((x-1)>=0&&visited[x-1]==false){
queue.add(Pair(x-1,number))
visited[x-1]=true
}
if((x+1)<=100000&&visited[x+1]==false){
queue.add(Pair(x+1,number))
visited[x+1]=true
}
if((x*2)<=100000&&visited[x*2]==false){
queue.add(Pair(x*2,number))
visited[x*2]=true
}
}
}
처음엔 그리디로 생각했다
처음 문제를 접했을 때, 순간이동(2*X)이 가장 빠르니까
무조건 순간이동을 최대한 많이 하는 방식이 최선일 거라고 생각했다.
예를 들어서 5 -> 10 -> 20 -> 21 -> 22 -> ... 식으로 동생 위치까지 다가가는 방식이다.
하지만 막상 몇 가지 테스트 케이스를 손으로 풀어보니,
- 중간에 걷는 게 더 빠른 경우도 있고
- 순간이동만 계속 쓰다 보면 엉뚱한 방향으로 멀어지는 경우도 생긴다
그래서 그리디 방식으로는 정확한 최소 시간을 구하기 어렵다는 걸 깨달았다.
BFS로 접근을 바꿨다
모든 경우의 수를 1초씩 늘려가며 탐색하면,
가장 먼저 동생 위치(K)에 도달했을 때의 시간이 곧 최소 시간이 된다.
그래서 이 문제는 너비 우선 탐색(BFS) 으로 푸는 게 정답이었다.
'코딩테스트 준비(kotlin) > DFS, BFS' 카테고리의 다른 글
[프로그래머스,kotlin] 네트워크 (0) | 2024.08.06 |
---|---|
[프로그래머스 kotlin] 숫자 변환하기 (0) | 2024.07.03 |
[프로그래머스, kotlin] 타겟넘버 (0) | 2024.06.28 |
[프로그래머스 , kotlin] 피로도 (0) | 2024.06.27 |
[kotlin] 백준 1280 DFS와 BFS (0) | 2023.10.27 |