【CodeForces】616C - The Labyrinth

http://codeforces.com/problemset/problem/616/C

给定n行m列的图=由 “*”和“.”构成的图。
题意求每个“”上下左右相连的“.”的个数包括“”本身。

预处理，否则会超时。

#include <bits/stdc++.h>
#include <set>
using namespace std;

int n,m;
int dx[4]={0,0,1,-1};
int dy[4]={1,-1,0,0};
int ans[1005][1005];
char g[1005][1005];
bool vis[1005][1005];
int t[1000002];
int sum;
int part=0;

void dfs(int x,int y){
    sum++;
    if (sum==10) sum=0;
    vis[x][y]=true;
    ans[x][y]=part;
    for (int i=0;i<4;i++){
        int xx=x+dx[i];
        int yy=y+dy[i];
        if (xx>=0&&xx<n&&yy>=0&&yy<m&&g[xx][yy]=='.'&&!vis[xx][yy]){
            dfs(xx,yy);
        }
    }
}

long long F(int x,int y){
    long long Ans=0;

    set <int> s;

    for (int i=0;i<4;i++){
        int xx=x+dx[i];
        int yy=y+dy[i];
        if (xx>=0&&xx<n&&yy>=0&&yy<m&&g[xx][yy]=='.'){
            s.insert(ans[xx][yy]);
        }
    }
    for (set<int>::iterator it=s.begin();it!=s.end();it++){
        Ans+=t[*it];
    }
    return Ans;
}
int main(){

    cin >> n >> m;
    for (int i=0;i<n;i++){
            scanf("%s",g[i]);
    }
    memset(vis,false,sizeof(vis));
    memset(t,0,sizeof(t));
    for (int i=0;i<n;i++){
        for (int j=0;j<m;j++){
            if (!vis[i][j]&&g[i][j]=='.'){
                sum=0;
                dfs(i,j);
                t[part++]=sum;
            }
        }
    }

    for (int i=0;i<n;i++){
        for (int j=0;j<m;j++){
            if (g[i][j]=='*'){
                printf("%lld",(F(i,j)+1)%10);
            }
            else printf(".");
        }
        printf("\n");
    }
}