题目描述
有一个仅由数字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
",cnt);
cnt=0;
}
}