[Baekjoon/C++] 7562번 - 나이트의 이동

[백준] Baekjoon Online Judge

문제로 이동

 

문제

체스판 위에 한 나이트가 놓여져 있다. 나이트가 한 번에 이동할 수 있는 칸은 아래 그림에 나와있다. 나이트가 이동하려고 하는 칸이 주어진다. 나이트는 몇 번 움직이면 이 칸으로 이동할 수 있을까?

 

입력

입력의 첫째 줄에는 테스트 케이스의 개수가 주어진다.

각 테스트 케이스는 세 줄로 이루어져 있다. 첫째 줄에는 체스판의 한 변의 길이 l(4 ≤ l ≤ 300)이 주어진다. 체스판의 크기는 l × l이다. 체스판의 각 칸은 두 수의 쌍 {0, ..., l-1} × {0, ..., l-1}로 나타낼 수 있다. 둘째 줄과 셋째 줄에는 나이트가 현재 있는 칸, 나이트가 이동하려고 하는 칸이 주어진다.

 

출력

각 테스트 케이스마다 나이트가 최소 몇 번만에 이동할 수 있는지 출력한다.

 

---

풀이

#include <iostream>
#include <string.h>
#include <queue>
using namespace std;

int bfs[301][301];
int dx[8] = { -2, -1, 1, 2, -2, -1, 1, 2 };
int dy[8] = { 1, 2, 2, 1, -1, -2, -2, -1 };

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

    int t;
    cin >> t;

    while (t--) {
        int n;
        cin >> n;

        queue <pair<int, int>> q;
        int startX, startY;
        int endX, endY;

        cin >> startX >> startY;
        cin >> endX >> endY;

        q.push({ startX, startY });

        while (!q.empty()) {
            int x = q.front().first;
            int y = q.front().second;
            int cnt = bfs[x][y];

            q.pop();

            if (x == endX && y == endY) {
                cout << cnt << '\n';
                break;
            }

            // 나이트 이동
            for (int i = 0; i < 8; i++) {
                int nx = x + dx[i];
                int ny = y + dy[i];

                // 나이트가 이동할 수 있는 위치인가?
                if (nx >= 0 && nx < n && ny >= 0 && ny < n && bfs[nx][ny] == 0) {
                    q.push({ nx, ny });
                    bfs[nx][ny] = cnt + 1;
                }
            }
        }

        // 배열 초기화
        memset(bfs, 0, sizeof(bfs));
    }

    return 0;
}

 

        q.push({ startX, startY });

        while (!q.empty()) {
            int x = q.front().first;
            int y = q.front().second;
            int cnt = bfs[x][y];

            q.pop();

            if (x == endX && y == endY) {
                cout << cnt << '\n';
                break;
            }

            // 나이트 이동
            for (int i = 0; i < 8; i++) {
                int nx = x + dx[i];
                int ny = y + dy[i];

                // 나이트가 이동할 수 있는 위치인가?
                if (nx >= 0 && nx < n && ny >= 0 && ny < n && bfs[nx][ny] == 0) {
                    q.push({ nx, ny });
                    bfs[nx][ny] = cnt + 1;
                }
            }
        }

 너비 우선 탐색을 하는 부분의 코드이다. 큐에 나이트의 위치를 push하고, 하나씩 꺼내서 그 위치에서 나이트가 갈 수 있는 장소를 계산한다. 이때 꺼낸 위치가 이동하려고 했던 최종 목적지라면 답을 출력하고 while문을 종료된다.

 계산한 위치가 체스판을 벗어나지 않았고, 처음 방문한 장소라면 큐에 위치 정보를 저장하고, bfs에 이동 횟수를 기록한다.