[programmers] 안전지대 - Java

문제 설명

제한사항

• board는 n * n 배열입니다.

• 1 ≤ n ≤ 100

• 지뢰는 1로 표시되어 있습니다.

• board에는 지뢰가 있는 지역 1과 지뢰가 없는 지역 0만 존재합니다.

입출력 예

입출력 예 설명

• (3, 2)에 지뢰가 있으므로 지뢰가 있는 지역과 지뢰와 인접한 위, 아래, 좌, 우, 대각선 총 8칸은 위험지역입니다. 따라서 16을 return합니다.

• (3, 2), (3, 3)에 지뢰가 있으므로 지뢰가 있는 지역과 지뢰와 인접한 위, 아래, 좌, 우, 대각선은 위험지역입니다. 따라서 위험지역을 제외한 칸 수 13을 return합니다.

• 모든 지역에 지뢰가 있으므로 안전지역은 없습니다. 따라서 0을 return합니다.

solution.java

class Solution {
    public int solution(int[][] board) {
        int n = board.length;
        int count = 0;

        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                if (board[i][j] == 1) {
                    markAdjacentCells(board, i, j);
                }
            }
        }

        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                if (board[i][j] == 0) {
                    count++;
                }
            }
        }

        return count;
    }

    private void markAdjacentCells(int[][] board, int row, int col) {
        int n = board.length;
        int[][] directions = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}, {-1, -1}, {-1, 1}, {1, -1}, {1, 1}};

        for (int[] dir : directions) {
            int newRow = row + dir[0];
            int newCol = col + dir[1];

            if (newRow >= 0 && newRow < n && newCol >= 0 && newCol < n && board[newRow][newCol] == 0) {
                board[newRow][newCol] = -1;
            }
        }
    }
}

핵심 키워드

• 이중 for문을 통해 2차원 배열을 순회하면서 읽어온 값이 1이라면 상,하,좌,우,대각선의 값을 -1로 설정한다.

• 만약 상,하,좌,우,대각선의 위치가 배열의 위치를 벗어날 경우, 해당 위치는 접근하지 않는다.

결론!