CodeForces 378C Maze (DFS)

题目链接

题意：给一个由“.”组成的联通区域，求再添加k个‘#'以后还是联通区域的方案。

分析：做题的时候犯二了，用DFS,一直搜到边缘，然后从边缘依次往回 回溯，回溯的过程中填充’#‘

一直填充k个。

因为在搜索的过程中，一直都是vis[][]标记的，所以时间复杂度最多只是搜了所有的边，即500*500*4.

DFS搜索的时间复杂度，取决于边的个数。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <queue>
#include <algorithm>
#define LL __int64
const int maxn = 500+10;
using namespace std;
char s[maxn][maxn];
int dx[] = {0, 0, 1, -1};
int dy[] = {1, -1, 0, 0};
int vis[maxn][maxn];
int n, m, cnt;
void dfs(int fx, int fy)
{
    vis[fx][fy] = 1;
    for(int i = 0; i < 4; i++)
    {
        int a, b;
        a = fx+dx[i];
        b = fy+dy[i];
        if(a>=1&&a<=n && b>=1&&b<=m){}
        else continue;
        if(s[a][b]!='.') continue;
        if(!vis[a][b])
        dfs(a, b);
    }
    if(cnt)
    {
        cnt --;
        vis[fx][fy] = 2;
    }
}

int main()
{
    int i, j;
    int fx, fy;
    while(~scanf("%d%d%d", &n, &m, &cnt))
    {
        memset(vis, 0, sizeof(vis));
        memset(s, 0, sizeof(s));
        for(i = 1; i <= n; i++)
        {
            getchar();
            for(j = 1; j <= m; j++)
            {
                scanf("%c", &s[i][j]);
                if(s[i][j]=='.')
                {
                    fx = i;
                    fy = j;
                }
            }
        }
        dfs(fx, fy);

        for(i = 1; i <= n; i++)
        {
            for(j = 1; j <= m; j++)
                {
                    if(vis[i][j]==2)
                    printf("X");
                    else
                    printf("%c", s[i][j]);
                }
            printf("
");
        }
    }
    return 0;
}