코딩테스트/프로그래머스

[프로그래머스] Java :: 안전지대

블로그 주인장 2024. 3. 30.

문제 링크

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

문제 설명

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

지뢰는 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

 

입출력 예 설명

  1. 입출력 예 #1 
    • (3, 2)에 지뢰가 있으므로 지뢰가 있는 지역과 지뢰와 인접한 위, 아래, 좌, 우, 대각선 총 8칸은 위험지역입니다. 따라서 16을 return합니다. 
  2. 입출력 예 #2 
    • (3, 2), (3, 3)에 지뢰가 있으므로 지뢰가 있는 지역과 지뢰와 인접한 위, 아래, 좌, 우, 대각선은 위험지역입니다. 따라서 위험지역을 제외한 칸 수 13을 return합니다. 
  3. 입출력 예 #3
    • 모든 지역에 지뢰가 있으므로 안전지역은 없습니다. 따라서 0을 return합니다.

 

코드 구현

import java.util.*;

class Solution {
    public int solution(int[][] board) {
        int answer = 0;
        int[] dx = {0,1,-1,0,-1,1,1,-1};
        int[] dy = {1,0,0,-1,-1,1,-1,1};
        List<int[]> list = new ArrayList<>();

        for(int i = 0; i < board.length; i++){
            for(int j = 0; j < board[0].length; j++){
                if(board[i][j] == 1){
                    list.add(new int[] {i, j});
                }
            }
        }

        for (int[] ints : list) {
            int i = ints[0];
            int j = ints[1];
            for (int k = 0; k < 8; k++) {
                if (dx[k] + i >= 0 && dy[k] + j >= 0
                        && dx[k] + i <= board.length - 1
                        && dy[k] + j <= board.length - 1) {
                    board[dx[k] + i][dy[k] + j] = 1;
                }
            }
        }

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

        return answer;
    }
}

 

풀이

  1. dx, dy의 값을 지뢰를 기준으로 좌표를 구해서 총 8방향을 도출할 수 있는 배열을 만든다.
  2. 지뢰의 위치를 미리 찾기 위해서 값이 1인 경우를 for문을 통해 찾고 List에 삽입한다.
  3. 지뢰의 위치를 담은 리스트에서 지뢰의 위치를 꺼내 dx, dy를 활용해서 8방향의 값을 1로 변경시킨다.
    • 즉, 8방향은 dx, dy에 각각 1을 더하고 빼서 실제 2차원 좌표에서 8방향을 이동하는 것처럼 사용한다.
  4. 이후 안전한 지대를 나타내는 0의 갯수를 세어 answer에 값을 넣어준 후에 반환해준다.
반응형

댓글