백준 숨바꼭질3 [c++]

13549 숨바꼭질3 골드5

문제

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

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

입력

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

출력

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

풀이

• 숨바꼭질1과 비슷한 형태이다.
• 위치에서 -1, +1이 될 때는 1초가 걸리는데, *2를 할때는 0초가 걸리는 점을 고려해서 위치와 pair가 되는 sec을 설정해주면 된다.

코드

#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
using namespace std;

int n, k;
int visited[100001];
queue<pair<int, int> > q;
vector<int> fast;

void bfs(int start)
{
    visited[start] = 1;
    q.push(make_pair(start, 0));

    while(!q.empty())
    {
        int pos = q.front().first;
        int sec = q.front().second;
        if(pos == k){
            fast.push_back(sec);
        }
        visited[pos] = 1;
        q.pop();

        if(pos - 1 >= 0 && pos - 1 <= 100000 && visited[pos - 1] == 0)
        {
            q.push(make_pair(pos - 1, sec + 1));
        }
        if(pos + 1 >= 0 && pos + 1 <= 100000 && visited[pos + 1] == 0)
        {
            q.push(make_pair(pos + 1, sec + 1));
        }
        if(pos * 2 >= 0 && pos * 2 <= 100000 && visited[pos * 2] == 0)
        {
            q.push(make_pair(pos * 2, sec));
        }
    }

}


int main()
{
    cin >> n >> k;
    
    bfs(n);
    sort(fast.begin(),fast.end());

    cout << fast[0] << endl;

    return 0;
}

• ps
• bfs