Algorithm/푼 문제

프로그래머스/Lv0/안전지대

안개바다 2023. 4. 1. 23:52

문제 설명

다음 그림과 같이 지뢰가 있는 지역과 지뢰에 인접한 위, 아래, 좌, 우 대각선 칸을 모두 위험지역으로 분류합니다.


지뢰는 2차원 배열 board에 1로 표시되어 있고 board에는 지뢰가 매설 된 지역 1과, 지뢰가 없는 지역 0만 존재합니다.
지뢰가 매설된 지역의 지도 board가 매개변수로 주어질 때, 안전한 지역의 칸 수를 return하도록 solution 함수를 완성해주세요.

제한 사항

  • board는 n * n 배열입니다.
  • 1 ≤ n ≤ 100
  • 지뢰는 1로 표시되어 있습니다.
  • board에는 지뢰가 있는 지역 1과 지뢰가 없는 지역 0만 존재합니다.

입출력 예

board result
[[0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 1, 0, 0], [0, 0, 0, 0, 0]] 16
[[0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 1, 1, 0], [0, 0, 0, 0, 0]] 13
[[1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1]] 0

board를 구조 분해해서 새로운 배열 result를 만들고( board의 값은 고정이어야 하므로 구조분해할당을 적용해 배열 result를 만들었다. ) board의 요소를 순회하면서 위험지역 들도 '1'로 값을 바꾸어 주었다. 이렇게 되면 안전지역은 '0' 위험 지역은 '1' 로 명확하게 구분된다.

단, 코너부분에서는 8방위 전부를 처리하면 에러가 나므로 예외처리를 적용해서 코드를 작성했다.

 

이후 result 배열의 '0'의 갯수( 안전 지역의 수 )를 count 해서 return했다.

function solution(board) {
    let result = Array.from({length: board.length}, () => Array.from({length: board.length}, () => 0));
    const n = board.length;
    for(let i=0; i<n; i++){
        for(let j=0; j<n; j++){
            if(board[i][j] === 1){
                result[i][j] = 1;
                if(i-1 >= 0 && j-1 >= 0) result[i-1][j-1] = 1;
                if(i-1 >= 0) result[i-1][j] = 1;
                if(i-1 >= 0 && j+1 < n) result[i-1][j+1] = 1;
                if(j-1 >= 0) result[i][j-1] = 1;
                if(j+1 < n) result[i][j+1] = 1;
                if(i+1 < n && j-1 >= 0) result[i+1][j-1] = 1;
                if(i+1 < n) result[i+1][j] = 1;
                if(i+1 < n && j+1 < n) result[i+1][j+1] = 1;
            }
        }
    }
    let count = 0;
    for (let a = 0; a < n; a++) {
        for (let b = 0; b < n; b++) {
            if (result[a][b] === 0) {
                count++;
            }
        }
    }
    return count;
}

처음 풀때는 이중 for문으로 해결했지만, reduce를 중첩해서 사용한다면 연산속도가 빨라질 것 이다.

 


다른 분의 풀이 중 깔끔하고 빠른 풀이가 있어서 참고했다.

function solution(board) {
    let outside = [[-1,0], [-1,-1], [-1,1], [0,-1],[0,1],[1,0], [1,-1], [1,1]];
    let safezone = 0;
    board.forEach((row, y, self) => row.forEach((it, x) => {
        if (it === 1) return false;
        return outside.some(([oy, ox]) => !!self[oy + y]?.[ox + x])
               ? false : safezone++;
    }));
    return safezone;
}