01迷宫(记忆化搜索)

来源:互联网 发布:linux系统下载安装jdk 编辑:程序博客网 时间:2024/05/16 09:21

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;}
1 0
原创粉丝点击