P1141 01迷宫(BFS+记忆化)

https://www.luogu.org/problemnew/show/P1141
题目描述
有一个仅由数字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
说明
所有格子互相可达。

对于20%的数据，n≤10；

对于40%的数据，n≤50；

对于50%的数据，m≤5；

对于60%的数据，n≤100,m≤100；

对于100%的数据，n≤1000,m≤100000。
Ac_code:
/*
询问次数特别多，数据比较大，考虑用BFS+记忆化搜索：
同一块连通块能抵达的格子是一样的，所以为每一块连通块标号，
同一块连通块上的点属于唯一的一个标号。然后记录每块连通块上格子总数，
那么询问的时候就直接可以根据询问点所属的连通块的标号直接输出ans;
*/

#include <iostream>
#include <stdio.h>
#define MAXN 1000005
using namespace std;
char a[1005][1005];
bool vis[1005][1005];
int p[MAXN];//记录每个格子所属的连通块编号
int ans[MAXN];//ans[x]存的是标号为x的连通块格子总数
int n,m,tno;//tno为连通块编号
int step[4][2]= {-1,0,1,0,0,-1,0,1};
struct point
{
    int x;
    int y;
} s,que[MAXN];//用que[MAXN]模拟队列，效率更高
bool check(point k,point tp)
{
    if(a[k.x][k.y]=='1')
    {
        if(tp.x>0&&tp.x<=n&&tp.y>0&&tp.y<=n&&!vis[tp.x][tp.y]&&a[tp.x][tp.y]=='0')
        {
            return true;
        }
        return false;
    }
    else
    {
        if(tp.x>0&&tp.x<=n&&tp.y>0&&tp.y<=n&&!vis[tp.x][tp.y]&&a[tp.x][tp.y]=='1')
        {
            return true;
        }
        return false;
    }
}
void BFS(point s)
{
    int head = 0,tail = 0;
    que[tail++] = s;
    vis[s.x][s.y] = true;
    int u = (s.x-1)*n + s.y;
    p[u] = tno;
    int cnt = 0;//记录当前连通块格子总数
    while(head < tail)
    {
        point k = que[head];
        cnt++;
        head++;
        point tp;
        for(int i = 0; i < 4; i++)
        {
            tp.x = k.x + step[i][0];
            tp.y = k.y + step[i][1];
            if(check(k,tp))
            {
                vis[tp.x][tp.y] = true;
                u = (tp.x-1)*n + tp.y;
                p[u] = tno;
                que[tail++] = tp;
            }
        }
    }
    ans[tno] = cnt;
}
int main()
{
    cin>>n>>m;
    tno = 0;
    for(int i = 1; i <= n; i++)
    {

        for(int j = 1; j <= n; j++)
        {
            scanf(" %c",&a[i][j]);
        }
    }
    for(int i = 1; i <= n; i++)
    {

        for(int j = 1; j <= n; j++)
        {
            if(!vis[i][j])
            {
                s.x = i;
                s.y = j;
                ++tno;
                BFS(s);
            }
        }
    }
    while(m--)
    {
        int x,y;
        cin>>x>>y;
        int pos = (x-1)*n + y;
        cout<<ans[p[pos]]<<endl;
    }
    return 0;
}