[프로그래머스][JAVA] 안전지대 (LV.0)

※ 문제 설명

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

 

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

 

※ 제한사항

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

 

※ 입출력 예

my_string	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

 

※ 입출력 예 설명

입출력 예 #1

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

입출력 예 #2

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

입출력 예 #3

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

 

---

나의 풀이

class Solution {
    public int solution(int[][] board) {
        int answer = 0;
        
        if(board.length == 1) {
            if(board[0][0] == 0) {
                answer = 1;
            } else {
                answer = 0;
            }
        } else {
            for(int i=0; i<board.length; i++) {
                for(int j=0; j<board[i].length; j++) {
                    if(board[i][j] == 1) {
                        if(j == 0 || j == board[i].length-1) {
                            if(j == 0) {
                                if(board[i][j+1] == 0) {
                                    board[i][j+1] = 2;
                                }
                            } else if(j == board[i].length-1) {
                                if(board[i][j-1] == 0) {
                                    board[i][j-1] = 2;
                                }
                            }
                        } else {
                            if(board[i][j+1] == 0) {
                                board[i][j-1] = 2;
                                board[i][j+1] = 2;
                            } else {
                                board[i][j-1] = 2;
                            }
                        }
                    }
                } 
            }

            for(int i=0; i<board.length; i++) {
                for(int j=0; j<board[i].length; j++) {
                    if(board[i][j] == 1 || board[i][j] == 2) {
                        if(i == 0 || i == board[i].length-1) {
                            if(i == 0) {
                                if(board[i+1][j] == 0) {
                                    board[i+1][j] = 3;
                                }
                            } else if(i == board[i].length-1) {
                                if(board[i-1][j] == 0) {
                                    board[i-1][j] = 3;
                                }
                            }
                        } else {
                            if(board[i-1][j] == 0) {
                                board[i-1][j] = 3;
                            } 
                            if(board[i+1][j] == 0) {
                                board[i+1][j] = 3;
                            }
                        }
                    }
                }
            }

            for(int i=0; i<board.length; i++) {
                for(int j=0; j<board[i].length; j++) {
                    if(board[i][j] == 0) {
                        answer++;
                    }
                }
            }
        }
        
        return answer;
    }
}

 

다른 사람의 풀이

public class Solution {
    int[][] d = {{-1, -1}, {-1, 0}, {-1, 1}, {0, -1}, {0, 0}, {0, 1}, {1, -1}, {1, 0}, {1, 1}};

    public int solution(int[][] board) {
        int n = board.length;
        boolean[][] unable = new boolean[n][n];
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                if (board[i][j] == 1) {
                    for (int[] k: d) {
                        int nearX = i + k[0];
                        int nearY = j + k[1];
                        if (nearX >= 0 && nearX < n && nearY >= 0 && nearY < n) {
                            unable[nearX][nearY] = true;
                        }
                    }
                } 
            }
        }

        int unableCount = 0;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                if (unable[i][j]) {
                    unableCount++;
                }
            }
        }
        return n * n - unableCount;
    }
}