CodeForces 598D Igor In the Museum （DFS+预处理）

题目链接：http://codeforces.com/problemset/problem/598/D

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

#define debug puts("YES");
#define rep(x,y,z) for(int (x)=(y);(x)<(z);(x)++)
#define ll long long

#define lrt int l,int r,int rt
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define root l,r,rt
#define mst(a,b) memset((a),(b),sizeof(a))
#define pii pair<ll,ll>
#define mk(x,y) make_pair(x,y)
const int  maxn =1e3+5;
const int mod=1e9+7;
const int ub=1e6;
ll powmod(ll x,ll y){ll t; for(t=1;y;y>>=1,x=x*x%mod) if(y&1) t=t*x%mod; return t;}
ll gcd(ll x,ll y){return y?gcd(y,x%y):x;}

/*
题目大意：给定一个地图，上面有墙和空白地，
空白地的权值就是相邻的墙的个数，
给定很多个询问，询问从给定点出发能得到的最大权值是多少。

题目分析：明显这道题询问次数比较恐怖，
要预处理下。对于每次搜索，我们都用dfs搜尽可达的所有点块
然后累计该连通块的答案，然后累加编号把地图进行编号，把编号映射成答案。

这样对于每次查询，我们都可以通过O（1）时间获得答案。
时间复杂度：O(n).
*/

int n,m,k,cur,x,y;
char s[maxn][maxn];
int mp[maxn][maxn];
int vis[maxn][maxn];
int ans[maxn*maxn];

int dx[]={-1,0,1,0};
int dy[]={0,-1,0,1};
int ret=0;
void dfs(int x,int y){
    if(vis[x][y]) return ;///
    vis[x][y]=cur;
    ret+=mp[x][y];
    rep(i,0,4){
        int tx=x+dx[i];
        int ty=y+dy[i];
        if(tx<0||tx>=n||ty<0||ty>=m) continue;
        if(mp[tx][ty]==-1) continue;
        dfs(tx,ty);
    }
}

int main(){
    scanf("%d%d%d",&n,&m,&k);
    mst(mp,-1);
    rep(i,0,n) scanf("%s",s[i]);
    rep(i,0,n) rep(j,0,m) if(s[i][j]=='.'){
        int t=0;
        if(i+1<n&&s[i+1][j]=='*') t++;
        if(i-1>=0&&s[i-1][j]=='*') t++;
        if(j+1<m&&s[i][j+1]=='*') t++;
        if(j-1>=0&&s[i][j-1]=='*') t++;
        mp[i][j]=t;
    }
    cur=1;
    rep(i,0,n) rep(j,0,m)
        if(mp[i][j]!=-1&&!vis[i][j]){
                cur++,ret=0;
                dfs(i,j);
                ans[cur]=ret;
        }
    rep(i,0,k){
        scanf("%d%d",&x,&y);x--,y--;
        printf("%d\n",ans[vis[x][y]]);
    }
    return 0;
}