[Baekjoon/C++] 13913번 - 숨바꼭질 4

[백준] Baekjoon Online Judge

문제로 이동

 

문제

수빈이는 동생과 숨바꼭질을 하고 있다. 수빈이는 현재 점 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>
using namespace std;

#define MAX 100001

vector<int> result;
int saveLine[MAX];
bool check[MAX];

void findPath(int start, int i);
void printPath();
void bfs(int start, int end);

// 방문 경로 구하기
void findPath(int start, int i) {
    while (i != start) {
        result.push_back(i);
        i = saveLine[i];
    }

    result.push_back(i);
}

// 방문 경로 출력하기
void printPath() {
    for (int i = result.size() - 1; i >= 0; i--)
        cout << result[i] << ' ';
}

void bfs(int start, int end) {
    queue<pair<int, int>> q;
    q.push({ start, 0 });
    check[start] = true;

    while (!q.empty()) {
        int x = q.front().first;
        int time = q.front().second;
        q.pop();

        // 동생이 있는 위치에 도달했음
        if (x == end) {
            cout << time << '\n';

            findPath(start, x);
            printPath();
        }

        int nx[3] = { x - 1, x + 1, x * 2 }; // 다음에 방문할 위치

        for (int i = 0; i < 3; i++) {
            if (nx[i] >= 0 && nx[i] < MAX && !check[nx[i]]) {
                q.push({ nx[i], time + 1 });

                check[nx[i]] = true;
                saveLine[nx[i]] = x;
            }
        }
    }
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);

    int N, K;
    cin >> N >> K;

    bfs(N, K);

    return 0;
}

너비 우선 탐색으로 문제를 풀었다.

 

void bfs(int start, int end) {
    queue<pair<int, int>> q;
    q.push({ start, 0 });
    check[start] = true;

    while (!q.empty()) {
        int x = q.front().first;
        int time = q.front().second;
        q.pop();

        // 동생이 있는 위치에 도달했음
        if (x == end) {
            cout << time << '\n';

            findPath(start, x);
            printPath();
        }

        int nx[3] = { x - 1, x + 1, x * 2 }; // 다음에 방문할 위치

        for (int i = 0; i < 3; i++) {
            if (nx[i] >= 0 && nx[i] < MAX && !check[nx[i]]) {
                q.push({ nx[i], time + 1 });

                check[nx[i]] = true;
                saveLine[nx[i]] = x;
            }
        }
    }
}

bfs를 하는 함수이다. 새로 어딘가를 방문하게 되면, check[]에 표시를 하고, saveLine[]에 바로 전에 방문한 위치를 저장하도록 했다.

 

// 방문 경로 구하기
void findPath(int start, int i) {
    while (i != start) {
        result.push_back(i);
        i = saveLine[i];
    }

    result.push_back(i);
}

saveLine[]으로 방문 경로를 구하는 함수이다. 해당 배열의 각 칸에는 바로 전에 방문한 칸의 번호가 들어 있다. 이를 이용하여 백터 배열에 숫자를 추가하도록 했다.

 

// 방문 경로 출력하기
void printPath() {
    for (int i = result.size() - 1; i >= 0; i--)
        cout << result[i] << ' ';
}

위에서 구한 방문 경로를 출력하는 함수이다. 위에서 구한 방문 경로는 역순으로 저장되어 있기에 출력은 맨 뒤부터 시작해야 한다.