codeforces 616C The Labyrinth (dfs)

链接：http://codeforces.com/contest/616/problem/C
题意：
给定一个矩阵，对于每个’*’的地方，假设将它变成 ‘.’ ， 那么要求她所在的全为’.’的连通块的大小。
思路：
很容易可能就是想到直接用dfs硬上，就跟那个经典的dfs的找水池数的题目类似，但是就会超时（别问我为什么知道的，我也侥幸想偷懒试过。。）
换一种想法，我们先不修改那个障碍的地方，直接先dfs预处理出原图中，每一个连通块的大小，并且将所有的连通块进行编号，记录下每一个点所在的连通块的标号。
那么最后我们枚举每一个’*’，对于它四周的所有’.’，将每个点所在的连通块的大小累加起来（这里要注意如果有多个点属于一个连通块的时候，这个连通块的大小只要加一次）最后再加上1。就是每一个点的答案了。
解题方法：
处理多个点属于同一个连通块的时候，可以使用set容器，set的特性本身就是有序的，没有重复的。然后遍历set中的元素。
代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <set>
using namespace std;
#define M 1009
bool vis[M][M];
int mark[M][M];
int dx[4] = {1,-1,0,0};
int dy[4] = {0,0,1,-1};
int n,m;
int id;
char mp[M][M];
int ans[M][M];
int cnt[M*10];
void dfs(int a,int b)
{
    mark[a][b] = id;
    cnt[id]++;
    for(int i = 0;i < 4;i++)
    {
        int x = a + dx[i];
        int y = b + dy[i];
        if(x < 0 || x >= n || y < 0 || y >= m || mark[x][y] || mp[x][y] == '*') continue; //如果已被标记则不进入
        dfs(x,y);
    }
}
int main()
{
    scanf("%d %d",&n,&m);
    id = 0;
    for(int i = 0;i < n;i++) scanf("%s",mp[i]);
    for(int i = 0;i < n;i++)
    {
        for(int j = 0;j < m;j++)
        {
            if(mp[i][j] == '.' && !mark[i][j]) //还没有被标记过
            {
                id++;
                dfs(i,j);
            }
        }
    }
    for(int i = 0;i < n;i++)
    {
        for(int j = 0;j < m;j++)
        {
            if(mp[i][j] == '*')
            {
                set<int> ss;
                for(int k = 0;k < 4;k++)
                {
                    int x = i + dx[k];
                    int y = j + dy[k];
                    if(x < 0 || x >= n || y < 0 || y >= m || mp[x][y] == '*') continue;
                    ss.insert(mark[x][y]);
                }
                set<int>::iterator it = ss.begin();
                int ans = 0;
                for(;it != ss.end();it++) ans += cnt[*it]; //遍历set中的元素
                ans++;
                printf("%d",ans%10);
            }
            else printf(".");
        }
        printf("\n");
    }
    return 0;
}