洛谷-1141 01迷宫

题目描述
有一个仅由数字0与1组成的n×n格迷宫。若你位于一格0上，那么你可以移动到相邻4格中的某一格1上，同样若你位于一格1上，那么你可以移动到相邻4格中的某一格0上。
你的任务是：对于给定的迷宫，询问从某一格开始能移动到多少个格子（包含自身）。
输入输出格式
输入格式：
第1行为两个正整数n,m。
下面n行，每行n个字符，字符只可能是0或者1，字符之间没有空格。
接下来m行，每行2个用空格分隔的正整数i,j，对应了迷宫中第i行第j列的一个格子，询问从这一格开始能移动到多少格。
输出格式：
m行，对于每个询问输出相应答案。

输入输出样例
输入样例#1：
2 2
01
10
1 1
2 2

输出样例#1：
4
4

说明
所有格子互相可达。
对于100%的数据，n≤1000,m≤100000

解释：这里我们遍历格子，把所有访问到的点用并查集给维护一下，最后直接查找祖先输出就好了

#include<cstdio>
#include<iostream>
using namespace std;
char G[1003][1003];
int dp[1003][1003]={0};
int father[1000003]={0};
int find(int x){
    if(father[x]==x) return x;
    return father[x]=find(father[x]);
}
void init(){
    for(int i=0;i<=1e6;i++) father[i]=i;
}
void merge(int x,int y){
    x=find(x);y=find(y);
    father[x]=y;
}
int n=0;
int dir[4][2]={0,1,0,-1,1,0,-1,0};
bool check(int x,int y){
    if(x<=n-1&&y<=n-1&&x>=0&&y>=0) return 1;
    return 0;
}
void dfs(int x,int y){
    dp[x][y]++;
    for(int i=0;i<4;i++){
        int dx=x+dir[i][0];
        int dy=y+dir[i][1];
        if(check(dx,dy)&&((G[x][y]-'0')+(G[dx][dy]-'0')==1)&&dp[dx][dy]==0){
            merge(n*dx+dy,x*n+y);
            dfs(dx,dy);
            dp[x][y]+=dp[dx][dy];
        }
    }
}
int m=0;
int main(){
    scanf("%d%d",&n,&m);
    init();
    for(int i=0;i<n;i++) cin>>G[i];
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            if(!dp[i][j]){
                dfs(i,j);
            }
        }
    }
    while(m--){
        int x=0,y=0;scanf("%d%d",&x,&y);
        x--;y--;
        int fa=find(x*n+y);
        printf("%d\n",dp[fa/n][fa%n]);
    }
    return 0;
}