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。
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
int n;
int next[4][2]={{0,-1},{1,0},{0,1},{-1,0}};
int book[1010][1010];
int str[1010][1010];
int ans[1010][1010];
int temp[1000001][2];
int cnt=0;
void dfs(int x,int y)
{
//if(book[x][y]==1)
//return; 不必要写
cnt++;
int i;
int xn,yn;
temp[cnt][0]=x;
temp[cnt][1]=y;
book[x][y]=1;
for(i=0;i<4;i++)
{
xn=x+next[i][0];
yn=y+next[i][1];
if(xn<=0||xn>n||yn<=0||yn>n)
continue;
if(book[xn][yn]==1)//回溯不重置为0,就是想只找一条深搜路径,找一条还是多条看题目的吧,笑
continue;
if(str[x][y]!=str[xn][yn])
dfs(xn,yn);
}
return;
}
int main()
{
int m;
int a,b;
scanf("%d%d",&n,&m);
memset(book,0,sizeof(book));
memset(ans,0,sizeof(ans));
int i,j;
for(i=1;i<=n;i++)
{
for(j=1;j<=n;j++)
scanf("%1d",&str[i][j]); //输入的精巧
}
for(i=1;i<=m;i++)
{
scanf("%d %d",&a,&b);
cnt=0;
if(book[a][b]==0)
{
dfs(a,b);
for(j=1;j<=cnt;j++)
ans[temp[j][0]][temp[j][1]]=cnt; //记忆化搜索,同一路径的所有点路程一样
}
printf("%d\n",ans[a][b]);
}
return 0;
}