BOJ - 13549 - 숨바꼭질 3

문제

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는 정수이다.

출력

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

구현

#include <iostream>
#include <deque>

using namespace std;

// map
const int MAX = 200004;
int visited[MAX];

int n, m, ans;

void bfs(int idx)
{
    deque<int> dq;
    dq.push_back(idx);
    visited[idx] = 1;
    while(dq.size())
    {
        int t = dq.front();
        dq.pop_front();
        if(t == m)
        {
            ans = visited[t];
            break;
        }
        if(!visited[t*2] && t*2 >= 0 && t*2 <= MAX)
        {
            visited[t*2] = visited[t];
            dq.push_front(t*2);
        }
        for(auto &el : {t+1, t-1})
            if(el >= 0 && el <= MAX)
                if(!visited[el])
                {
                    visited[el] = visited[t] + 1;
                    dq.push_back(el);
                }
    }
}

int main()
{
    cin >> n >> m;
    bfs(n);
    cout << ans-1 << endl;
    return 0;
}

SOL

1. map 즉, 그래프를 구현하되 조건을 방문하지 않고, 최댓값을 넘기지 않았을 경우 bfs 함수를 호출해 준다.
2. deque로 우선순위 큐를 만들어 주어 조건의 우선순위로 두어야 하는 순간이동 부분의 긴능을 우선시하게 한다.

Reference & 다른 PS 모음집

GitHub - ggh-png/PS: 2021.01.11 start

BFS 너비 우선 탐색