[백준 알고리즘] 2665번: 미로만들기

https://www.acmicpc.net/problem/2665

문제

문제

n×n 바둑판 모양으로 총 n2개의 방이 있다. 일부분은 검은 방이고 나머지는 모두 흰 방이다. 검은 방은 사면이 벽으로 싸여 있어 들어갈 수 없다. 서로 붙어 있는 두 개의 흰 방 사이에는 문이 있어서 지나다닐 수 있다. 윗줄 맨 왼쪽 방은 시작방으로서 항상 흰 방이고, 아랫줄 맨 오른쪽 방은 끝방으로서 역시 흰 방이다.

시작방에서 출발하여 길을 찾아서 끝방으로 가는 것이 목적인데, 아래 그림의 경우에는 시작방에서 끝 방으로 갈 수가 없다. 부득이 검은 방 몇 개를 흰 방으로 바꾸어야 하는데 되도록 적은 수의 방의 색을 바꾸고 싶다.

아래 그림은 n=8인 경우의 한 예이다.

미로찾기

위 그림에서는 두 개의 검은 방(예를 들어 (4,4)의 방과 (7,8)의 방)을 흰 방으로 바꾸면, 시작방에서 끝방으로 갈 수 있지만, 어느 검은 방 하나만을 흰 방으로 바꾸어서는 불가능하다. 검은 방에서 흰 방으로 바꾸어야 할 최소의 수를 구하는 프로그램을 작성하시오.

단, 검은 방을 하나도 흰방으로 바꾸지 않아도 되는 경우는 0이 답이다.

입력

첫 줄에는 한 줄에 들어가는 방의 수 n(1≤n≤50)이 주어지고, 다음 n개의 줄의 각 줄마다 0과 1이 이루어진 길이가 n인 수열이 주어진다. 0은 검은 방, 1은 흰 방을 나타낸다.

출력

첫 줄에 흰 방으로 바꾸어야 할 최소의 검은 방의 수를 출력한다.

풀이

다익스트라 알고리즘으로 풀 수 있는 문제였다. 각 좌표를 초기에는 모두 INT_MAX 로 초기화해두고 bfs 로 좌표를 탐색하면서 해당 좌표까지 도달하는데 필요한 제거 블록수를 최소값으로 업데이트한다. 오버플로우를 막기위해 최소값이 되지 않는 이동은 큐에 포함시키지 않는다.

코드

#include <cstdio>
#include <vector>
#include <queue>
#include <climits>

using namespace std;
int n;
vector<vector<bool>> map;
vector<vector<int>> dist;

bool isInBound(int row, int col) {
    return row >= 1 && row <= n && col >= 1 && col <= n;
}

void bfs() {
    int minChange = n * n;
    priority_queue<pair<int, pair<int, int>>> pq;

    pq.push({ 0, {1, 1} });

    int rowDir[] = { 1, 0, -1, 0 };
    int colDir[] = { 0, 1, 0, -1 };

    while (!pq.empty()) {
        pair<int, pair<int, int>> now = pq.top();
        pq.pop();
        for (int i = 0; i < 4; i++) {
            int nextRow = now.second.first + rowDir[i];
            int nextCol = now.second.second + colDir[i];

            if (isInBound(nextRow, nextCol)
                && map[nextRow][nextCol] == 1
                && now.first < dist[nextRow][nextCol]) {
                dist[nextRow][nextCol] = now.first;
                pq.push({ now.first,{ nextRow, nextCol} });
            }
            else if (isInBound(nextRow, nextCol)
                && map[nextRow][nextCol] == 0
                && now.first + 1 < dist[nextRow][nextCol]) {
                dist[nextRow][nextCol] = now.first + 1;
                pq.push({ now.first + 1,{ nextRow, nextCol} });
            }
        }
    }
}

int main() {
    scanf("%d", &n);
    map.resize(n + 1, vector<bool>(n + 1));
    dist.resize(n + 1, vector<int>(n + 1, INT_MAX));

    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= n; j++) {
            int point;
            scanf("%1d", &point);
            map[i][j] = (bool)point;
        }
    }

    bfs();
    printf("%d", dist[n][n]);
}

전여훈
Written by@전여훈 (Click Me!)
고민이 담긴 코드를 만들자, 고민하기 위해 공부하자.