[c++] BOJ 1600 :: 말이 되고픈 원숭이

난이도 : ?

걸린 시간 : 1시간 반

 

문제

동물원에서 막 탈출한 원숭이 한 마리가 세상구경을 하고 있다. 그 녀석은 말(Horse)이 되기를 간절히 원했다. 그래서 그는 말의 움직임을 유심히 살펴보고 그대로 따라 하기로 하였다. 말은 말이다. 말은 격자판에서 체스의 나이트와 같은 이동방식을 가진다. 다음 그림에 말의 이동방법이 나타나있다. x표시한 곳으로 말이 갈 수 있다는 뜻이다. 참고로 말은 장애물을 뛰어넘을 수 있다.

	x		x
x				x
		말
x				x
	x		x

근데 원숭이는 한 가지 착각하고 있는 것이 있다. 말은 저렇게 움직일 수 있지만 원숭이는 능력이 부족해서 총 K번만 위와 같이 움직일 수 있고, 그 외에는 그냥 인접한 칸으로만 움직일 수 있다. 대각선 방향은 인접한 칸에 포함되지 않는다.

이제 원숭이는 머나먼 여행길을 떠난다. 격자판의 맨 왼쪽 위에서 시작해서 맨 오른쪽 아래까지 가야한다. 인접한 네 방향으로 한 번 움직이는 것, 말의 움직임으로 한 번 움직이는 것, 모두 한 번의 동작으로 친다. 격자판이 주어졌을 때, 원숭이가 최소한의 동작으로 시작지점에서 도착지점까지 갈 수 있는 방법을 알아내는 프로그램을 작성하시오.

 

풀이

• 최단 경로 문제이므로 bfs
• 경로가 없을 수도 있음 주의

 

코드

#include <iostream>
#include <queue>

using namespace std;

struct Node {
    int r, c, k, path;
    Node(int row, int col, int num, int p) {
        r = row; c = col; k = num; path = p;
    }
};

int direction[4][2] = { {0,1}, {1,0}, {0, -1}, {-1, 0} };
int horseDirection[8][2] = { {1,2}, {2, 1}, {2, -1}, {1, -2}, {-1, -2}, {-2, -1}, {-2, 1}, {-1, 2} };

int map[201][201];
int memo[201][201][31]; // 메모이제이션 => visited 배열

int main() {
    // 10시 10분 시작
    // 말의 움직임으로 k번, 나머지는 인접한 칸
    // 0 : 평지, 1: 장애물
    // k는 30이하
    // 앞으로 2 옆으로 1
    // bfs (해당 위치에서 k를 썼는지 안썼는지 메모이제이션)
    // (r,c,k)

    int K; cin >> K;
    int W, H; cin >> W >> H;

    fill(&map[0][0], &map[200][201], -1);
    fill(&memo[0][0][0], &memo[200][200][30], -1);

    for (int i = 0; i < H; i++) {
        for (int j = 0; j < W; j++) {
            cin >> map[i][j];
        }
    }

    int result = 1e8;
    queue<Node*> que;
    que.push(new Node(0, 0, K, 0));
    while (!que.empty()) {
        Node* pos = que.front(); que.pop();
        int path = pos->path + 1; // 넣지 않고 time 전역변수로 사용하면 됨
        if (pos->r == H - 1 && pos->c == W - 1) {
            // 마지막에 도착
            result = result > path - 1 ? path - 1 : result;
            break;
        }


        for (int i = 0; i < 4; i++) {
            int r = pos->r + direction[i][0];
            int c = pos->c + direction[i][1];

            if (r<0 || c<0 || r>=H || c>=W) // 맵 밖에 위치
                continue;
            if (map[r][c] == 1) // 장애물이 있음
                continue;
            if (memo[r][c][pos->k] != -1) // 이미 들름 => 사실 k 이하의 모든 memo가 있으면 안해도 됨
                continue;
            que.push(new Node(r, c, pos->k, path));
            memo[r][c][pos->k] = 0;
        }

        if (pos->k == 0)
            continue;

        for (int i = 0; i < 8; i++) {
            int r = pos->r + horseDirection[i][0];
            int c = pos->c + horseDirection[i][1];

            if (r < 0 || c < 0 || r >= H || c >= W) // 맵 밖에 위치
                continue;
            if (map[r][c] == 1) // 장애물이 있음
                continue;
            if (memo[r][c][pos->k-1] != -1) // 이미 들름 => 넣을 때 체크해줘야 메모리 초과 x
                continue;
            que.push(new Node(r, c, pos->k-1, path)); // K 사용
            memo[r][c][pos->k-1] = 0;
        }
    }
    if (result == 1e8) {
        result = -1;
    }
    cout << result;
}

 

주의할 점

• bfs 시 time 변수를 전역으로 사용하여 1씩 증가시키는 방식으로 하는 것이 Node struct에 넣는 것보다 편하다.
• 최종 결과를 만족했을 때에는 최단 경로이므로 그냥 break

 

결과

아이디	메모리	시간	언어	코드 길이
gmldms784	45488	144	C++17 / 수정	2270