BOJ - 13913 - 숨바꼭질 4

문제

13913번: 숨바꼭질 4

문제

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

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

입력

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

출력

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

둘째 줄에 어떻게 이동해야 하는지 공백으로 구분해 출력한다.

구현

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

// trace
const int MAX = 200005;
int trace[MAX];
int visited[MAX];

int n, m;

vector<int> v;

void bfs(int idx)
{
    queue<int> q;
    q.push(idx);
    visited[idx] = 1;
    trace[idx] = 1;
    while(q.size())
    {
        int t = q.front();
        q.pop();
        if(t == m)
        {
            cout << visited[t] - 1 << endl;
            break;
        }
        for(auto &el : {t-1, t+1, t*2})
            if(el >= 0 && el < MAX)
                if(!visited[el])
                {
                    q.push(el);
                    trace[el] = t;
                    visited[el] = visited[t] + 1;
                }
    }
    for(int i=m; i != n; i = trace[i])
        v.push_back(i);
    v.push_back(n);
    reverse(v.begin(), v.end());
}

int main()
{
    cin >> n >> m;
    bfs(n);
    
    for(auto &el : v)
        cout << el << " ";
    cout << endl;
    return 0;
}

SOL

1. map 즉, 그래프를 구현하되 조건을 방문하지 않고, 최댓값을 넘기지 않았을 경우 bfs 함수를 호출해 준다.

2. trace 배열을 만들어 최단경로의 흔적을 표현한다.

       // 시작지점부터 도착지점이 아니면 반복 
   		for(int i=m; i != n; i = trace[i])
           v.push_back(i);
       v.push_back(n);

Reference & 다른 PS 모음집

GitHub - ggh-png/PS: 2021.01.11 start

BFS 너비 우선 탐색