洛谷 P1141 01迷宫（DFS）

                                                P1141 01迷宫

题目链接： https://www.luogu.org/problemnew/show/P1141

题目描述

有一个仅由数字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。

题解：用一个数组记录当前点可以走到的每个点，那么这些点能走到的点数都是相同的，然后每个点DFS就行了。

AC代码：

#include <bits/stdc++.h>
using namespace std;
#define _for(i,a,b) for(int i=a;i<=b;i++)
const int maxn = 1006;
int n,m,now,leng[1000007][2],vis[maxn][maxn],ans[maxn][maxn];
char a[maxn][maxn];
int xx[]={1,-1,0,0},yy[]={0,0,1,-1};
void bfs(int x,int y)
{
	now++;
	leng[now][0]=x;
	leng[now][1]=y;
	_for(i,0,3)
	{
		int nowx = x+xx[i];
		int nowy = y+yy[i];
		if(nowx<=n&&nowx>=1&&nowy<=n&&nowy>=1&&vis[nowx][nowy]==0&&a[x][y]!=a[nowx][nowy])
			{
				vis[nowx][nowy]=1;
				bfs(nowx,nowy);
			}
	}
	return;
}
int main(int argc, char const *argv[])
{
	cin>>n>>m;
	_for(i,1,n)
	{
		_for(j,1,n)cin>>a[i][j];
	}
	_for(i,1,n)
	{
		_for(j,1,n)
		{
			now = 0;
			if(vis[i][j]==0)
				{	
					vis[i][j]=1;
					bfs(i,j);
					_for(k,1,now)ans[leng[k][0]][leng[k][1]]=now;
				}
		}
	}
	_for(i,1,m)
	{
		int x,y;
		cin>>x>>y;
		cout<<ans[x][y]<<endl;
	}
	return 0;
}