[BFS广度优先搜索] 迷宫

描述

给定一个仅由数字 0 与 1 组成的 n 行 m 列的迷宫。若你位于一格 0 上，那么你可以移动到相邻 4 格中的任意一格 1 上；同样地，若你位于一格 1 上, 那么你可以移动到相邻 4 格中的任意一格 0 上。

现在，有 q 次询问。每次询问从给定的某一格 (第 x 行，第 y 列，均从 0 开始标号) 开始，在规则内任意移动。这样，有一些格子是可能到达的，而其余格子是无法到达的。求可能到达的格子数量。

输入

输入第一行包含两个整数 n 和 m（1 ≤ n, m ≤ 103）。
接下来 n 行，每行包含 m 个 0 或 1 的整数，表示迷宫的内容。
接下来一行包含一个整数 q（0 ≤ q ≤ 105）。
接下来 q 行，每行包含两个整数 x 和 y，表示一次询问（0 ≤ x < n，0 ≤ y < m）。

输出

输出 q 行，每行一个整数，依次表示每个询问的答案。

样例输入

5 4
1 1 0 1
1 1 1 1
1 1 0 0
0 0 1 1
0 1 0 0
4
4 3
0 1
3 0
1 1

样例输出

4
11
2
1

提示

对于第二条询问，如图所示，从黄色格子 (0, 1) 出发，可分别到达所有绿色格子，包含自身共 11 格。

 解题分析

这道题目首先可以看出是一道很经典的BFS广度优先搜索的题目。我们只需要用一个队列，一个visited数组，一些规则与判断就可以比较容易地去遍历我们能够到达的所有格子。但是很显然，如果我们每一个格子都这样去遍历的话，最后询问次数一旦多了起来，就很容易超时。所以我们想到了一个可以用一个数组记录下来的方法，这个时候我们还要想到其实按照题目所述的做法，我们这样是对称的。也就是说这是一个“连通块”，我们只需要找出这些连通块，每次遍历一遍后把所有走过的连通块都标记上相同的步数即可。为了更加便捷快速，我们可以用一个巧妙的自定义结构体Node创建一个数组，然后记录下我们走过的格子的位置路径，一路保存下来然后最后直接用一个循环去遍历更新即可。最后，我们的输入输出就变简单了，如果是我们已经遍历更新过的位置，那么我们直接输出路径的长度（或者说是遍历的所有格子的数量），如果是我们还没有遍历的位置，我们就调用bfs函数遍历一遍再输出我们的答案即可。

代码实现

#include <iostream>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#include <cstring>
#include <string>
#include <cstdio>
// #include <iomanip>
#include <map>
#include <stack>
#include <list>
#include <set>
#include <unordered_map>
#include <unordered_set>
#include <queue>
#include <sstream>
#define INF (1<<30)

using namespace std;

struct node{
	int x,y;
} f[10000100];

int n,m;
char maze[1001][1001]={0};
int record[1001][1001]={0};
bool visited[1001][1001]={0};
int dx[]={0,0,1,-1};
int dy[]={1,-1,0,0};

void bfs(int x,int y){
	queue<node> q;
	q.push({x,y});
	visited[x][y]=1;
	node tmp;
	int t=0;
	while(!q.empty()){
		tmp=q.front();
		q.pop();
		if(record[tmp.x][tmp.y]) continue;
		for(int i=0;i<4;i++){
			int nx = tmp.x + dx[i] , ny = tmp.y + dy[i];
			if(nx <= 0 || ny <= 0 || nx > n || ny > n) continue;
			if(visited[nx][ny] || record[nx][ny]) continue;
			if(maze[nx][ny]==maze[tmp.x][tmp.y]) continue;
			q.push({nx,ny});
			visited[nx][ny]=1;
			f[t].x=nx; f[t].y=ny;
			t++;
		}
	}
	record[x][y]=t+1;
	for(int i=0;i<t;i++){
		record[f[i].x][f[i].y]=t+1;
	}
}


int main(){
    scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
			scanf("%s",maze[i]+1);
		}
	int tmpx,tmpy;
	for(int i=0;i<m;i++){
		scanf("%d%d",&tmpx,&tmpy);
		if(record[tmpx][tmpy]) printf("%d\n",record[tmpx][tmpy]);
		else {
			bfs(tmpx,tmpy);
			printf("%d\n",record[tmpx][tmpy]);
		}
	}
    return 0;
}