본문 바로가기
코딩테스트/자바 Level 0

[Java] 안전지대

by onggury 2023. 6. 30.

문제

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

지뢰는 2차원 배열 board 에 1로 표시되어 있고 board 에는 지뢰가 매설 된 지역 1과, 지뢰가 없는 지역 0만 존재합니다.

지뢰가 매설된 지역의 지도 board 가 매개변수로 주어질 때, 안전한 지역의 칸 수를 return 하도록 solution 함수를 완성해주세요.

 

 

제한사항

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

 

import java.util.*;

class Solution {
    public int solution(int[][] board) {
        int[] y = {-1, -1, 0, 1, 1, 1, 0, -1};
        int[] x = {0, 1, 1, 1, 0, -1, -1, -1};
        
        int n = board.length;
        int answer = n*n;
        
        int[][] danger = new int[n][];
        for(int i = 0; i < n; i++) {
            danger[i] = board[i].clone();
        }
        
        for(int i = 0; i < n; i++) {
            for(int j = 0; j < n; j++) {
                if(board[i][j] == 1) {
                    danger[i][j] = 1;
                    answer -= 1;
                    int cy = i;
                    int cx = j;
                    for(int loc = 0; loc < 8; loc++) {
                        int ny = cy + y[loc];
                        int nx = cx + x[loc];
                        if(ny < 0 || ny >= n || nx < 0 || nx >= n || danger[ny][nx] == 1) {
                            continue;
                        }
                        danger[ny][nx] = 1;
                        answer -= 1;
                    }
                }
            }
        }
        return answer;
    }
}

board 배열만 이용해서 문제를 해결하려고 했으나 위험지역을 1로 변경하는 바람에 원래 0이였던 값이 1로 바뀌다보니 배열 순회 시 그 지역도 지뢰가 매설되어있다고 판단하여 잘못된 결과를 반환하였다.

1 말고 다른 값으로 바꾸자니 인근 지뢰 매설지역까지 신경써서 값을 바꿔야 하는 경우때문에 그냥 danger 라는 배열을 새로 만들었다. 그리고 거기에 지뢰 매설 좌표 포함 총 9개의 칸을 1로 바꾸는 방향으로 해결했다.

최종 결과는 '전체 원소의 개수(danger은 전부 0이였으니 0의 개수) - 1로 바뀐 개수(danger의 원소가 0 -> 1로 바뀐 개수)' 로 계산했다.

 

 

출처

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