[백준] 1600번 - 말이 되고픈 원숭이 [Java]

https://www.acmicpc.net/problem/1600

 

[백준] 1600번 - 말이 되고픈 원숭이 [Java]

[백준] 1600번 - 말이 되고픈 원숭이 [Java]

---

1.  아이디어

 

각 이동 방법에 대한 2가지 탐색 배열과 방문 체크처리를 통해 해결할 수 있다.

---

2. 문제풀이

 

각 위치에서 원숭이의 이동 방법과 말의 이동 방법 모두 가능하므로 2가지 이동 방법에 대한 탐색 배열을 각각 만들어서 각각 반복문으로 새 위치를 생성해서 사용했다. 원숭이의 이동 방법의 경우 일반적인 4방 탐색으로 진행하며 말의 이동 방법은 해당 위치를 담은 8방 탐색으로 진행했다. 방문 체크의 경우 말의 이동 방법을 몇 번 사용해서 온 위치인지까지 기록해야해서 3차원 boolean 타입 배열을 통해 위치와 이동 방법까지 저장한 방법과 2차원 int 타입 배열로 위치에 이동 방법을 저장한 두 가지 케이스로 해결했다.

---

3. 코드

 

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

public class Main {

    private static class Node {
        int r;
        int c;
        int jump;

        public Node(int r, int c, int jump) {
            this.r = r;
            this.c = c;
            this.jump = jump;
        }
    }

    private static final int[] dr = {-1, 0, 1, 0};
    private static final int[] dc = {0, 1, 0, -1};
    private static final int[] drh = {-1, -2, -2, -1, 1, 2, 2, 1};
    private static final int[] dch = {-2, -1, 1, 2, 2, 1, -1, -2};

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;

        int K = Integer.parseInt(br.readLine());
        st = new StringTokenizer(br.readLine());

        int W = Integer.parseInt(st.nextToken());
        int H = Integer.parseInt(st.nextToken());

        int[][] map = new int[H][W];
        for (int i = 0; i < H; i++) {
            st = new StringTokenizer(br.readLine());

            for (int j = 0; j < W; j++) {
                map[i][j] = Integer.parseInt(st.nextToken());
            }
        }

        int dist = bfs(H, W, K, map);
        System.out.println(dist);
    }

    private static int bfs(int H, int W, int K, int[][] map) {
        Queue<Node> q = new ArrayDeque<>();
        q.offer(new Node(0, 0, 0));

        boolean[][][] visited = new boolean[H][W][1 + K];
        visited[0][0][0] = true;

        int dist = 0;

        while (!q.isEmpty()) {
            int len = q.size();

            for (int i = 0; i < len; i++) {
                Node node = q.poll();
                if (node.r == H - 1 && node.c == W - 1) return dist;

                // 원숭이의 이동 방법
                for (int dir = 0; dir < 4; dir++) {
                    int nr = node.r + dr[dir];
                    int nc = node.c + dc[dir];

                    if (nr < 0 || nr >= H || nc < 0 || nc >= W) continue;
                    if (map[nr][nc] == 1 || visited[nr][nc][node.jump]) continue;

                    q.offer(new Node(nr, nc, node.jump));
                    visited[nr][nc][node.jump] = true;
                }

                // 말의 이동 방법
                for (int dir = 0; dir < 8; dir++) {
                    int nr = node.r + drh[dir];
                    int nc = node.c + dch[dir];

                    if (nr < 0 || nr >= H || nc < 0 || nc >= W) continue;
                    if (node.jump == K) continue;
                    if (map[nr][nc] == 1 || visited[nr][nc][node.jump + 1]) continue;

                    q.offer(new Node(nr, nc, node.jump + 1));
                    visited[nr][nc][node.jump + 1] = true;
                }
            }

            dist++;
        }

        return -1;
    }

}

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

public class Main {

    private static class Node {
        int r;
        int c;
        int jump;

        public Node(int r, int c, int jump) {
            this.r = r;
            this.c = c;
            this.jump = jump;
        }
    }

    private static final int[] dr = {-1, 0, 1, 0};
    private static final int[] dc = {0, 1, 0, -1};
    private static final int[] drh = {-1, -2, -2, -1, 1, 2, 2, 1};
    private static final int[] dch = {-2, -1, 1, 2, 2, 1, -1, -2};

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;

        int K = Integer.parseInt(br.readLine());
        st = new StringTokenizer(br.readLine());

        int W = Integer.parseInt(st.nextToken());
        int H = Integer.parseInt(st.nextToken());

        int[][] map = new int[H][W];
        for (int i = 0; i < H; i++) {
            st = new StringTokenizer(br.readLine());

            for (int j = 0; j < W; j++) {
                map[i][j] = Integer.parseInt(st.nextToken());
            }
        }

        int dist = bfs(H, W, K, map);
        System.out.println(dist);
    }

    private static int bfs(int H, int W, int K, int[][] map) {
        Queue<Node> q = new ArrayDeque<>();
        q.offer(new Node(0, 0, 0));

        int[][] visited = new int[H][W];
        for (int i = 0; i < H; i++) {
            Arrays.fill(visited[i], Integer.MAX_VALUE);
        }
        visited[0][0] = 0;

        int dist = 0;

        while (!q.isEmpty()) {
            int len = q.size();

            for (int i = 0; i < len; i++) {
                Node node = q.poll();
                if (node.r == H - 1 && node.c == W - 1) return dist;

                // 원숭이의 이동 방법
                for (int dir = 0; dir < 4; dir++) {
                    int nr = node.r + dr[dir];
                    int nc = node.c + dc[dir];

                    if (nr < 0 || nr >= H || nc < 0 || nc >= W) continue;
                    if (map[nr][nc] == 1 || visited[nr][nc] <= node.jump) continue;

                    q.offer(new Node(nr, nc, node.jump));
                    visited[nr][nc] = node.jump;
                }

                // 말의 이동 방법
                for (int dir = 0; dir < 8; dir++) {
                    int nr = node.r + drh[dir];
                    int nc = node.c + dch[dir];

                    if (nr < 0 || nr >= H || nc < 0 || nc >= W) continue;
                    if (node.jump == K) continue;
                    if (map[nr][nc] == 1 || visited[nr][nc] <= node.jump + 1) continue;

                    q.offer(new Node(nr, nc, node.jump + 1));
                    visited[nr][nc] = node.jump + 1;
                }
            }

            dist++;
        }

        return -1;
    }

}

---

4. 후기

 

- 3차원 boolean 타입 배열

- 2차원 int 타입 배열