1. 程式人生 > 實用技巧 >cf337A Maze - dfs - 連通塊刪除k個取餘還要連通

cf337A Maze - dfs - 連通塊刪除k個取餘還要連通

給一個連通塊,刪除k個塊,使得剩下的塊還是連通的

因為是一個連通塊,直接dfs一次即可,同時記錄一下dfs當前的塊座標。
然後轉置下座標的順序,刪除前k個即可
原理:
對於一個點,如果能從這個點進行dfs,那麼也就是說,可以從這個點向多個方向走,如果刪除這個點,那麼可能造成不連通。
但是對於dfs越往後的點,是單方向的,可以刪

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
typedef long long lint;
typedef pair<int, int> pi;
int n, m, s;
char str[555][555];
int dx[4] = {1, 0, -1, 0}, dy[4] = {0, 1, 0, -1};
bool vis[555][555];
vector<pi> ord;
void dfs(int x, int y){
    if(vis[x][y] || str[x][y] == '#') return;
    vis[x][y] = 1;
    ord.push_back(pi(x, y));
    for(int i = 0; i < 4; i++){
        if(x + dx[i] < 0 || x + dx[i] >= n || y + dy[i] < 0 || y + dy[i] >= m) continue;
        dfs(x + dx[i], y + dy[i]);
    }
}

int main(){
    cin >> n >> m >> s;
    for(int i = 0; i < n; i++) cin >> str[i];
    for(int i = 0; i < n; i++){
        for(int j = 0; j < m; j++){
            if(str[i][j] == '.'){
                dfs(i, j);
                reverse(ord.begin(), ord.end());
                for(int k = 0; k < s; k++){
                    str[ord[k].first][ord[k].second] = 'X';
                }
                for(int i = 0; i < n; i++) cout << str[i] << endl;
                return 0;
            }
        }
    }
}