Codeforces 377 A Maze【DFS】

题意：给出n*m的矩阵，矩阵由'.'和'#'组成，再给出k，表示需要在'.'处加k堵墙，使得剩下的'.'仍然是连通的

先统计出这个矩阵里面总的点数'.'为sum

因为题目说了一定会有一个解，所以找到一个'.',就开始搜，搜到sum-k个连通的点的时候跳出，剩下的没有访问过的点就全部填成墙

#include<iostream>  
#include<cstdio>  
#include<cstring> 
#include <cmath> 
#include<stack>
#include<vector>
#include<map> 
#include<set>
#include<queue> 
#include<algorithm>  
using namespace std;

typedef long long LL;
const int INF = (1<<30)-1;
const int mod=1000000007;
const int maxn=1005;

char g[maxn][maxn];
int vis[maxn][maxn];
int ans,n,m,k,cnt;
int dir[4][2]={1,0,-1,0,0,1,0,-1};

void dfs(int x,int y){
    vis[x][y]=1;
    cnt++;
    if(cnt==ans) return;
    for(int i=0;i<4;i++){
        if(cnt==ans) return;
        int xx=x+dir[i][0];
        int yy=y+dir[i][1];
        if(xx<1||xx>n||yy<1||yy>m||vis[xx][yy]||g[xx][yy]!='.') continue;
        dfs(xx,yy);
    }    
}

int main(){
    scanf("%d %d %d",&n,&m,&k);
    for(int i=1;i<=n;i++)
     for(int j=1;j<=m;j++) cin>>g[i][j];
     
      ans=0;
     for(int i=1;i<=n;i++){
         for(int j=1;j<=m;j++){
             if(g[i][j]=='.') ans++;
         }
     }
     ans=ans-k;
     
     for(int i=1;i<=n;i++){
         int flag=0;
         for(int j=1;j<=m;j++){
             cnt=0;
             if(g[i][j]=='.'){
                 dfs(i,j);
                 flag=1;
                 break;                 
             }
         }
         if(flag) break;
     }
     
     for(int i=1;i<=n;i++){
         for(int j=1;j<=m;j++){
             if(g[i][j]=='.'&&!vis[i][j]) g[i][j]='X';
         }
     }
     
     
     for(int i=1;i<=n;i++){
         for(int j=1;j<=m;j++)
         printf("%c",g[i][j]);
         printf("
");
     }
     return 0;
}