[프로그래머스] 리코쳇 로봇 - Java

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/169199

문제

리코쳇 로봇이라는 보드게임이 있습니다.

이 보드게임은 격자모양 게임판 위에서 말을 움직이는 게임으로, 시작 위치에서 목표 위치까지 최소 몇 번만에 도달할 수 있는지 말하는 게임입니다.

이 게임에서 말의 움직임은 상, 하, 좌, 우 4방향 중 하나를 선택해서 게임판 위의 장애물이나 맨 끝에 부딪힐 때까지 미끄러져 이동하는 것을 한 번의 이동으로 칩니다.

다음은 보드게임판을 나타낸 예시입니다.

...D..R
.D.G...
....D.D
D....D.
..D....

여기서 "."은 빈 공간을, "R"은 로봇의 처음 위치를, "D"는 장애물의 위치를, "G"는 목표지점을 나타냅니다.
위 예시에서는 "R" 위치에서 아래, 왼쪽, 위, 왼쪽, 아래, 오른쪽, 위 순서로 움직이면 7번 만에 "G" 위치에 멈춰 설 수 있으며, 이것이 최소 움직임 중 하나입니다.

게임판의 상태를 나타내는 문자열 배열 board가 주어졌을 때, 말이 목표위치에 도달하는데 최소 몇 번 이동해야 하는지 return 하는 solution함수를 완성하세요. 만약 목표위치에 도달할 수 없다면 -1을 return 해주세요.

제한 사항

알고리즘: BFS (너비 우선 탐색)

풀이

기본 BFS 문제이다.
너비 우선 탐색을 진행해서 먼저 G에 도달한다면 그 노드의 c가 최솟값이다.

소스 코드

import java.util.*;
class Point {
    int x,y, c;
    Point(int x, int y, int c) {
        this.x = x;
        this.y = y;
        this.c = c;
    }
}
class Solution {
    static final int[] dx = {-1, 0, 1, 0};
    static final int[] dy = {0, -1, 0, 1};
    static Point start, end;
    static boolean[][] visited;
    static int W,H;
    public int solution(String[] board) {
        H = board.length;
        W = board[0].length();
        visited = new boolean[H][W];
        for(int i=0; i<board.length; i++) {
            for(int j=0; j<board[i].length(); j++) {
                if(board[i].charAt(j) == 'R') start = new Point(j, i, 0);
                else if(board[i].charAt(j) == 'G') end = new Point(j, i, 0);
            }
        }
        int answer = BFS(board);
        return answer;
    }
    static int BFS(String[] board) {
        Queue<Point> que = new LinkedList<>();
        que.add(start);
        visited[start.y][start.x] = true;
        while(que.size()!=0) {
            Point n = que.poll();
            if((n.y == end.y) && (n.x == end.x)) return n.c;
            for(int i=0; i<4; i++) {
                Point np = move(i, new Point(n.x, n.y, n.c + 1), board);
                if(!visited[np.y][np.x]) {
                    visited[np.y][np.x] = true;
                    que.add(np);
                }
            }
        }
        return -1;
    }
    static Point move(int dir, Point p, String[] board) {
        while(true) {
            int nx = p.x + dx[dir];
            int ny = p.y + dy[dir];
            if(((0<=nx && nx<=W-1) && (0<=ny && ny<=H-1)) && board[ny].charAt(nx) != 'D') {
                p.x = nx;
                p.y = ny;
            } else break;
        }
        return p;
    }
}