프로그래머스 [Lv. 2] 리코쳇 로봇 {언어 : JavaScript}

문제 링크

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

프로그래머스

정답 코드

function solution(board) {
    const n = board.length;
    const m = board[0].length;
    
    const dr = [-1, 1, 0, 0];
    const dc = [0, 0, -1, 1];
    const isValid = (nr, nc) => {
        return 0 <= nr && nr < n && 0 <= nc && nc < m;
    };
    
    let [Rr, Rc] = [0, 0];
    let [Gr, Gc] = [0, 0];
    let cnt = 0;
    for (let i = 0; i < n; i++) {
        if (cnt == 2) {
            break;
        };
        for (let j = 0; j < m; j++) {
            if (cnt == 2) {
                break;
            };
            const boardIJ = board[i][j];
            if (boardIJ == "R") {
                cnt += 1;
                [Rr, Rc] = [i, j];
                continue;
            };
            if (boardIJ == "G") {
                cnt += 1;
                [Gr, Gc] = [i, j];
            };
        };
    };
    
    q = [[Rr, Rc, 0]];
    v = Array(n).fill().map(() => Array(m).fill().map(() => 10001));
    let isPossible = false;
    while (q.length) {
        const [r, c, l] = q.shift();
        if (r == Gr && c == Gc) {
            isPossible = true;
            break;
        };
        for (let d = 0; d < 4; d++) {
            const [drd, dcd] = [dr[d], dc[d]];
            let nr = r + drd;
            let nc = c + dcd;
            if (isValid(nr, nc) && board[nr][nc] != "D") {
                while (isValid(nr+drd, nc+dcd) && board[nr+drd][nc+dcd] != "D") {
                    nr += drd;
                    nc += dcd;
                };
                if (v[nr][nc] > l + 1) {
                    v[nr][nc] = l + 1;
                    q.push([nr, nc, l+1]);
                };
            };
        };
    };
    const answer = isPossible ? v[Gr][Gc] : -1;
    return answer;
}

풀이 방법

1. 로봇과 도착점의 위치를 찾는다.
2. 방문배열을 최대 반복 횟수 대략 10001로 측정하고 n * m 배열로 만든다.
3. 그래프를 순회하면서 도착한 지점과 비교해서 현재 값이 작으면 방문배열에 저장한다.
4. 도착지점에 도달할 수 있으면 v[Gr][Gc] 값을, 아니면 -1을 출력한다.

느낀점

• 모르면 답을 보고 어떻게 푸는지 확인하면서 많은 문제를 접하는 것이 좋다.