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;
}