洛谷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

说明

所有格子互相可达。

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

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

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

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

对于100%的数据，n≤1000，m≤100000。

这道题宽搜只能70分。。。
所以用宽搜+记忆化（不是纯记忆化，只是思路一样）
代码还是很好理解的，注意不能用memset，否则会TLE

#include<bits/stdc++.h>
int n,m,a[1010025][3],cnt,aa[100005],k;
int x[4]={0,1,0,-1},y[4]={1,0,-1,0},d[1005][1005];
bool b[1005][1005];
inline void bfs(int i,int j){
    cnt++;d[i][j]=k;
    int head=1,foot=2;
    a[2][1]=i,a[2][2]=j;
    while(head<foot){
        head++;
        for(register int q=0;q<=3;q++){
            int xx=a[head][1]+x[q];
            int yy=a[head][2]+y[q];
            if(xx>0 && xx<=n && yy>0 && yy<=n && !d[xx][yy]
            && b[a[head][1]][a[head][2]]!=b[xx][yy]){
                    cnt++;
                    foot++;
                    a[foot][1]=xx;
                    a[foot][2]=yy;
                    d[xx][yy]=k;
               }
        }
    }
    aa[k]=cnt;
}
int main(){
    scanf("%d%d",&n,&m);
    for(register int i=1;i<=n;i++)
        for(register int j=1;j<=n;j++){
            char c;
            scanf(" %c",&c);
            if(c=='1')  b[i][j]=1;
        }
    for(register int i=1;i<=m;i++){
        int xxx,yyy;
        scanf("%d%d",&xxx,&yyy);
        k++;
        if(d[xxx][yyy])   cnt=aa[d[xxx][yyy]];
        else   bfs(xxx,yyy);
        printf("%d\n",cnt);
        cnt=0;
    }
}