백준 16946번 벽 부수고 이동하기 4 C++

출처 : https://www.acmicpc.net/problem/16946

 

벽이 있는 위치에서 상하좌우를 검사하면서 빈 공간이 얼마나 이어져있는지 구해야 한다.

그런데 매번 bfs로 검사하면 시간이 오래 걸리기 때문에 한번 방문해서 이어진 곳은 따로 표시를 해둬서 다음번에 방문할때는 표시해둔 값만 더해주면 된다.

 

그리고 주의해야할점이 하나 더 있는데, 만약 벽에서 위쪽 빈공간과 오른쪽 빈공간이 이어져있다면 거리를 두번 더하기 때문에 이 부분을 잘 처리해야 한다.

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

using namespace std;
using int64 = long long;

int n, m;

int Hash = 0;

int dRow[4] = { -1,0,1,0 };
int dCol[4] = { 0,1,0,-1 };

vector<vector<pair<int, int>>> cache;
vector<vector<int>> board;
vector<vector<bool>> found;

void Bfs(int row, int col) {
    int cnt = 1;
    vector<int> hashes;
    for (int i = 0; i < 4; i++) {
        vector<pair<int, int>> points;
        int nextRow = row + dRow[i];
        int nextCol = col + dCol[i];
        if (nextRow < 0 || nextRow >= n) continue;
        if (nextCol < 0 || nextCol >= m) continue;
        if (board[nextRow][nextCol] != 0) continue;
        if (cache[nextRow][nextCol].second != -1) {
            if (find(hashes.begin(), hashes.end(), cache[nextRow][nextCol].first) == hashes.end()) {
                hashes.push_back(cache[nextRow][nextCol].first);
                cnt += cache[nextRow][nextCol].second;
            }
            continue;
        }
        int _cnt = 0;
        queue<pair<int, int>> q;
        q.push({ nextRow,nextCol });
        found[nextRow][nextCol] = true;
        while (!q.empty()) {
            auto curr = q.front();
            q.pop();
            points.push_back(curr);
            _cnt++;
            for (int j = 0; j < 4; j++) {
                int _nextRow = curr.first + dRow[j];
                int _nextCol = curr.second + dCol[j];
                if (_nextRow < 0 || _nextRow >= n) continue;
                if (_nextCol < 0 || _nextCol >= m) continue;
                if (board[_nextRow][_nextCol] != 0) continue;
                if (found[_nextRow][_nextCol]) continue;
                q.push({ _nextRow,_nextCol });
                found[_nextRow][_nextCol] = true;
            }
        }
        hashes.push_back(Hash);
        for (auto a : points) {
            cache[a.first][a.second].first = Hash;
            cache[a.first][a.second].second = _cnt;
        }
        Hash++;
        cnt += _cnt;
    }
    board[row][col] = cnt;
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    cin >> n >> m;
    board = vector<vector<int>>(n, vector<int>(m));
    cache = vector<vector<pair<int, int>>>(n, vector<pair<int, int>>(m, { -1,-1 }));
    found = vector<vector<bool>>(n, vector<bool>(m, false));
    for (int i = 0; i < n; i++) {
        string str;
        cin >> str;
        for (int j = 0; j < m; j++) {
            board[i][j] = str[j] - '0';
        }
    }
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            if (board[i][j] == 0)
                continue;
            Bfs(i, j);
        }
    }
    for (auto a : board) {
        for (auto b : a) {
            cout << b % 10;
        }
        cout << "\n";
    }
}