[코딩테스트 Lv.0] 안전지대

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

출처 : 프로그래머스

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

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

입출력 예 #3
모든 지역에 지뢰가 있으므로 안전지역은 없습니다. 따라서 0을 return합니다.

---

내 풀이

[Java]

import java.util.*;
class Solution {
    public int solution(int[][] board) {
        int answer = 0;
        int x[] = {-1, 1, 0, 0, -1, -1, 1, 1};
        int y[] = {0, 0, -1, 1, -1, 1, -1, 1};
        
        ArrayList<int[]> b = new ArrayList<>();
        
        for(int i = 0; i < board.length; i++){
            for(int j = 0; j < board[i].length; j++){
                if(board[i][j] == 1){
                    int tmp[] = {i, j};
                    b.add(tmp);
                }
            }
        }
        
        for(int i = 0; i < b.size(); i++){
            int m = b.get(i)[0];
            int n = b.get(i)[1];
            for(int j = 0; j < x.length; j++){
                if(
                    x[j] + m >= 0 
                   && y[j] + n >= 0
                   && x[j] + m < board.length
                   && y[j] + n < board.length
                  ){
                    board[x[j] + m][y[j] + n] = 2;
                }                
            }
        }
        
        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;
    }
}

 

[C]

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>

// board_rows는 2차원 배열 board의 행 길이, board_cols는 2차원 배열 board의 열 길이입니다.
int solution(int** board, size_t board_rows, size_t board_cols) {
    int answer = 0;
    int x[8] = {-1, 1, 0, 0, -1, -1, 1, 1};
    int y[8] = {0, 0, -1, 1, -1, 1, -1, 1};
    
       
    for(int i = 0; i < board_rows; i++){
        for(int j = 0; j < board_cols; j++){
            if(board[i][j] == 1){
                for(int k = 0; k < 8; k++){
                    if(x[k] + i >= 0 && y[k] + j >= 0 && x[k] + i < board_rows && y[k] + j < board_cols ){
                        if(board[x[k] + i][y[k] + j] == 0){
                            board[x[k] + i][y[k] + j] = 2;
                        }
                    }
                }
            }
        }
    }
    
    for(int i = 0; i < board_rows; i++){
        for(int j = 0; j < board_cols; j++){
            if(board[i][j] == 0) answer++;
        }
    }
    
    return answer;
}

 

[Python]

def solution(board):
    answer = 0
    x = [-1, 1, 0, 0, -1, -1, 1, 1]
    y = [0, 0, -1, 1, -1, 1, -1, 1]
    
    b = []
    
    for i in range(len(board)):
        for j in range(len(board[i])):
            if board[i][j] == 1:
                tmp = [i, j]
                b.append(tmp)
        
    for i in range(len(b)):
        m = b[i][0]
        n = b[i][1]
        for j in range(8):
            if x[j] + m >= 0 and y[j] + n >= 0 and x[j] + m < len(board) and y[j] + n < len(board):
                board[x[j] + m][y[j] + n] = 2
    
    for i in range(len(board)):
        for j in range(len(board[i])):
            if board[i][j] == 0:
                answer += 1
    
    return answer

 

---

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

프로그래머스