[프로그래머스/C++ 문제 풀이] 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, 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
[[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

 

입출력 예 #1

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

입출력 예 #2

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

입출력 예 #3

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

---

문제 풀이

이번 문제는 간단히 생각해서 지뢰가 위치한 곳과 해당 위치를 기준으로 8방향을 제외한 남은 공간의 수를 구하는 문제입니다.

 

저는 지뢰의 위치(y, x)들을 저장하는 배열을 만들고, 모든 위치를 순회하며 8방향을 전부 1의 값으로 변화시켰습니다. 또한 이 과정에서 현재 위험 구역(지뢰 or 지뢰의 8방향)의 수를 체크하여 전체 보드의 수(n * n)에서 빼는 방식으로 문제를 해결하였습니다.

 

> 8방향 위치를 쉽게 순회하기 위한 변화량 배열(dy, dx)

// 8방향 위치 배열 (상, 우상, 우, 우하, 하, 좌하, 좌, 좌상 순서)
int dy[8] = { 1, 1, 0, -1, -1, -1, 0, 1 };
int dx[8] = { 0, 1, 1, 1, 0, -1, -1, -1 };

 

#include <string>
#include <vector>

using namespace std;

// 8방향 위치 배열 (상, 우상, 우, 우하, 하, 좌하, 좌, 좌상 순서)
int dy[8] = { 1, 1, 0, -1, -1, -1, 0, 1 };
int dx[8] = { 0, 1, 1, 1, 0, -1, -1, -1 };

int solution(vector<vector<int>> board) {
    int answer = 0;

    // 지뢰가 매설된 위치를 저장하기 위한 배열 (y, x의 위치 배열)
    vector<pair<int, int>> loc;

    // 위험 지역의 수를 누적하기 위한 변수
    int risk = 0;

    for (int y = 0; y < board.size(); y++)
    {
        for (int x = 0; x < board.size(); x++)
        {
            // 폭탄이 존재한다면?
            if (board[y][x] == 1)
            {
                // 위치 배열에 해당 좌표 넣기
                loc.push_back(make_pair(y, x));
            }
        }
    }

    // 위치 배열에 들어있는 지뢰의 수만큼 위험 지역이라고 표시하기
    risk += loc.size();

    // 위치 배열에 들어있는 지뢰의 위치를 순회하며 주변 8방향 위험 지역(1)으로 설정하기
    for (int i = 0; i < loc.size(); i++)
    {
        for (int j = 0; j < 8; j++)
        {
            int ny = loc[i].first + dy[j];
            int nx = loc[i].second + dx[j];

            // 만약 해당 위치가 보드의 범위를 벗어난다면 - 건너뛰기
            if (nx < 0 || ny < 0 || nx >= board.size() || ny >= board.size()) continue;
            // 만약 해당 위치가 이미 지롸가 있거나, 위험 지역이라면? - 건너뛰기
            if (board[ny][nx] == 1) continue;

            // 위험 지역(1)으로 설정하기
            board[ny][nx] = 1;
            // 위험 지역의 수 증가시키기
            risk++;
        }
    }

    // 전체 지역의 수에서 위험 지역의 수 빼기
    answer = board.size() * board.size() - risk;

    return answer;
}