[백준] 1600번 : 말이 되고픈 원숭이 (Java)

🔗 1600번 : 말이 되고픈 원숭이
● Java 풀이
● 시간 제한 2초
● 메모리 제한 256 MB

 

문제

 

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

	x		x
x				x
		말
x				x
	x		x

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

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

 

테스트 케이스

/** 출처 (https://www.acmicpc.net/board/view/137559) **/

// input
3
10 4
0 1 0 0 1 1 1 1 1 1
0 1 0 1 1 0 1 1 1 1
0 0 0 1 1 1 1 0 1 1
1 1 1 1 1 1 1 1 1 0

// answer
10

// input
1
7 5
0 1 0 0 0 0 0
0 1 0 1 1 1 0
0 1 0 1 0 0 0
0 0 0 1 1 1 1
1 1 1 1 1 0 0

// answer
16

// input
1
4 4
0 1 1 1
0 0 1 1
1 0 1 1
1 1 1 0

// answer
4

// input
5
5 8
0 0 0 0 0
1 1 1 1 0
1 1 1 1 0
0 0 1 1 0
0 1 1 0 0
0 1 1 1 1
0 1 1 1 1
0 0 0 0 0

// answer
11

// input
5
6 8
0 0 0 0 0 0
1 1 1 1 1 0
1 1 1 1 1 0
0 0 0 1 1 0
0 1 1 1 0 0
0 1 1 1 1 1
0 1 1 1 1 1
0 0 0 0 0 0

// answer
14

// input
5
7 8
0 0 0 0 0 0 0
1 1 1 1 1 1 0
1 1 1 1 1 1 0
0 0 0 1 1 1 0
0 1 1 1 0 0 0
0 1 1 1 1 1 1
0 1 1 1 1 1 1
0 0 0 0 0 0 0

// answer
17

// input
5
7 8
0 0 0 0 0 0 0
1 1 1 1 1 1 0
1 1 1 1 1 1 0
0 0 0 1 1 1 0
0 1 1 1 0 0 0
0 1 0 1 1 1 1
0 1 1 1 1 1 1
0 0 0 0 0 0 0

// answer
13

// input
5
8 8
0 0 0 0 0 0 0 0
1 1 1 1 1 1 1 0
1 1 1 1 1 1 1 0
0 0 0 0 1 1 1 0
0 1 1 1 1 0 0 0
0 1 1 1 1 1 1 1
0 1 1 1 1 1 1 1
0 0 0 0 0 0 0 0

// answer
20

// input
2
2 6
0 1
1 1
1 0
1 1
0 1
0 0

// answer
4

// input
2
10 2
0 0 1 0 0 1 0 0 1 0
0 0 1 1 0 0 0 0 1 0

// answer
10

// input
2
9 2
0 0 1 1 0 0 1 1 0
0 0 1 0 0 0 1 0 0

// answer
7

// input
1
5 3
0 1 0 0 0
0 1 0 1 0
0 0 0 1 0

// answer
6

// input
1
1 1
0

// answer
0

// input
2
10 2
0 0 1 0 0 1 0 0 1 0
0 0 1 1 0 0 0 0 1 0

// answer
10

// input
1
5 5
0 1 1 0 1
0 0 1 0 1
0 1 0 1 1
0 1 0 1 0
1 1 0 1 0

// answer
-1

// input
2
5 3
0 0 0 0 0
1 0 1 1 0
1 0 1 1 0

// answer
4

// input
5
6 6
0 0 0 0 0 1
0 0 0 1 0 1
0 1 0 0 0 1
0 1 0 0 1 0
0 0 0 0 0 1
1 0 0 0 1 0

// answer
4

// input
1
4 4
0 0 0 0
0 0 0 0
0 0 1 1
0 0 1 0

// answer
4

 

풀이

package BJ01000;

import java.io.*;
import java.util.*;

public class BJ01600_말이되고픈원숭이 {
    private static class Pos {
        public Pos(int r, int c, int k) {
            this.r = r;
            this.c = c;
            this.k = k;
        }

        int r, c, k;
    }

    private static int K, W, H;  // 나이트식 이동 제한 횟수, 가로 세로 길이
    private static long[][][] map;
    private static final int INF = 20001;
    private static final int[] dr = {-1, 1, 0, 0}, dc = {0, 0, -1, 1};  // 원숭이 이동 경로
    private static final int[] kr = {-2, 2, -2, 2, -1, 1, -1, 1}, kc = {-1, 1, 1, -1, -2, 2, 2, -2};  // 말 이동 경로

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        K = Integer.parseInt(br.readLine());
        StringTokenizer st = new StringTokenizer(br.readLine());
        W = Integer.parseInt(st.nextToken());
        H = Integer.parseInt(st.nextToken());
        map = new long[H][W][K + 1];  // 세로, 가로, 말 이동 횟수 3자원배열 생성

        for (int i = 0; i < H; i++) {
            st = new StringTokenizer(br.readLine());
            for (int j = 0; j < W; j++) {
                Arrays.fill(map[i][j], st.nextToken().equals("1") ? -1 : INF);  // 값을 전부 inf 아니면 장애물로 채우기
            }
        }
        Arrays.fill(map[0][0], 0);  // 시작점은 항상 0

        moveMap();  // 이동 위치 찾는 함수 호출

        long minMove = map[H - 1][W - 1][0];  // 최소 이동 횟수 선언 및 정의
        for (int i = 1; i <= K; i++) {
            minMove = Math.min(minMove, map[H - 1][W - 1][i]);  // 오른쪽 끝까지의 최소 이동 횟수 계산
        }

        System.out.println(minMove == INF ? -1 : minMove);  // 최소 이동 횟수 출력 (이동 못할 시 -1)

    }

    private static void moveMap() {
        Queue<Pos> queue = new ArrayDeque<>();
        queue.add(new Pos(0, 0, 0));  // 큐에 첫 이동점 삽입

        Pos pos;
        int nr, nc;
        while (!queue.isEmpty()) {  // BFS
            pos = queue.poll();
            for (int i = 0; i < 4; i++) {  // 원숭이용 이동
                nr = pos.r + dr[i];
                nc = pos.c + dc[i];  // 이동 위치 계산

                if (0 <= nr && nr < H && 0 <= nc && nc < W && map[pos.r][pos.c][pos.k] + 1 < map[nr][nc][pos.k]) {  // 이동 가능할 경우 (이동 횟수보다 작은 경우)
                    map[nr][nc][pos.k] = map[pos.r][pos.c][pos.k] + 1;  // 최소 이동 횟수 갱신
                    queue.add(new Pos(nr, nc, pos.k));  // Queue에 삽입 (새로운 인접 이동거리 계산을 위해)
                }
            }

            for (int i = 0; i < 8 && pos.k < K; i++) {  // 나이트용 이동
                nr = pos.r + kr[i];
                nc = pos.c + kc[i];  // 이동 위치 계산

                if (0 <= nr && nr < H && 0 <= nc && nc < W && map[pos.r][pos.c][pos.k] + 1 < map[nr][nc][pos.k + 1]) {  // 이동 가능할 경우 (이동 횟수보다 작은 경우)
                    map[nr][nc][pos.k + 1] = map[pos.r][pos.c][pos.k] + 1;  // 최소 이동 횟수 갱신
                    queue.add(new Pos(nr, nc, pos.k + 1));  // Queue에 삽입 (새로운 인접 이동거리 계산을 위해)
                }
            }
        }
    }

    private static void printMap(int k) {
        for (int i = 0; i < H; i++) {
            for (int j = 0; j < W; j++) {
                System.out.printf("%3d ", map[i][j][k]);
            }
            System.out.println();
        }
        System.out.println();
    }
}

 

역경의 SQLD 시험이 끝난 후 오랜만에 푸는 골드 3 문제 ㅠㅠ 오히려 그리웠다.

의외로 너무 답이 보였고, 너무 쉽게 풀려서 더 좋았다.

테스트 케이스를 제공해주신 분도 감사하게도 한 번에 많은 테스트 케이스를 주셔서 블로그 정리도 편했다는 사실!

 

여기서 틀릴 수 있는 점은 2가지 정도로 추릴 수 있을 거 같다.

1. 말 이동 가능 횟수 갱신
2. 말 이동 이후 큐에 삽입

나는 비록 INF 값을 말 이동 횟수 생각 안하고 401로 두어서 자꾸 3퍼센트에서 틀렸지만....

말 이동 횟수가 좌측 상단으로 갈 경우만 생각해도 이동 경로를 큐에 삽입해줘야한다는 것을 알 수 있다.

그리고, 이동 횟수가 제한이 있으므로 이를 구분 지은 DP를 생성해야 한다.

그 외에는 방문 체크 해주면서 BFS 돌려주면 될 것 같다.