01迷宫(记忆化搜索)
01迷宫
题目描述:
有一个由01组成的n*n格迷宫,若你位于一格0上,那么你可以移动到相邻4格中的某一格1上,同样若你位于一格1上,那么你可以移动到相邻4格中的某一格0上。那么对于给定的迷宫,询问从某一格开始能移动到多少格。
输入描述:
输入的第1行为两个正整数n,m。
下面n行,每行n个字符,字符只可能是0或者1,字符之间没有空格。
接下来m行,每行2个用空格分隔的正整数i,j,对应了迷宫中第i行第j列的一个0,询问从这一格开始能移动到多少格。
输出描述:
输出包括m行,对于每个询问输出答案。
样例输入:
2 2
01
10
1 1
2 2
样例输出:
4
4
数据范围及提示:
样例说明:
所有格子互相可达。
数据规模:
对于20%的数据,n≤10;
对于40%的数据,n≤50;
对于50%的数据,m≤5;
对于60%的数据,n≤100,m≤100;
对于100%的数据,n≤1000,m≤1000000。
思路:
因为有多个查询,为了减少不必要的搜索,使用记忆化搜索
#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
const int maxn=1010;
int n,m,map[maxn][maxn],me[maxn][maxn];
int xx[5]={0,1,-1,0,0},
yy[5]={0,0,0,1,-1};
char s[maxn];
struct node
{
int x;
int y;
}a;
struct tong
{
int x;
int y;
}b[maxn*maxn];
bool flag[maxn][maxn];
queue<node> q;
void search(node a)
{
if(me[a.x][a.y])
{
printf("%d\n",me[a.x][a.y]);
return;
}
int cnt=0;
q.push(a);
while(!q.empty())
{
a=q.front();
q.pop();
for(int i=1;i<=4;i++)
{
node t;
t.x=a.x+xx[i];
t.y=a.y+yy[i];
if(t.x>=1&&t.x<=n&&t.y>=1&&t.y<=n&&!flag[t.x][t.y]&&map[a.x][a.y]+map[t.x][t.y]==1)
{
flag[t.x][t.y]=1;
cnt++;
b[cnt].x=t.x;
b[cnt].y=t.y;
q.push(t);
}
}
}
for(int i=1;i<=cnt;i++)
me[b[i].x][b[i].y]=cnt;
if(cnt==0) cnt++;
printf("%d\n",cnt);
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
{
cin>>s;
for(int j=0;j<n;j++)
map[i][j+1]=s[j]-'0';
}
for(int i=1;i<=m;i++)
{
scanf("%d%d",&a.x,&a.y);
search(a);
}
return 0;
}