P1141 01迷宫——bfs解法

题目描述

有一个仅由数字000与111组成的n×nn \times nn×n格迷宫。若你位于一格0上，那么你可以移动到相邻444格中的某一格111上，同样若你位于一格1上，那么你可以移动到相邻444格中的某一格000上。

你的任务是：对于给定的迷宫，询问从某一格开始能移动到多少个格子（包含自身）。
输入格式

第111行为两个正整数n,mn,mn,m。

下面nnn行，每行nnn个字符，字符只可能是000或者111，字符之间没有空格。

接下来mmm行，每行222个用空格分隔的正整数i,ji,ji,j，对应了迷宫中第iii行第jjj列的一个格子，询问从这一格开始能移动到多少格。
输出格式

mmm行，对于每个询问输出相应答案。
输入输出样例
输入 #1

2 2
01
10
1 1
2 2

输出 #1

4
4

说明/提示

所有格子互相可达。

对于20%20%20%的数据，n≤10n≤10n≤10；

对于40%40%40%的数据，n≤50n≤50n≤50；

对于50%50%50%的数据，m≤5m≤5m≤5；

对于60%60%60%的数据，n≤100,m≤100n≤100,m≤100n≤100,m≤100；

对于100%100%100%的数据，n≤1000,m≤100000n≤1000,m≤100000n≤1000,m≤100000。

在一开始看到这道题的时候，想用广度优先搜索，处理每一个输入的位置，暴力解决。但是提交后发现有三个点超时了。
看了题解后发现要使用计算连通图的大小进行求解。
连通图思路：每一块连通图中的连通度都相同，所以可以一开时就对输入的地图进行处理，标记好每一块连通图和对应的连通度，然后在处理时直接查找就好。
代码实现：

#include <iostream>
#include <iomanip>
#include <cstring>
using namespace std;

struct note
{
	int x, y;
}que[1000001]; //结构体数组，和head，tail一起实现队列

char map[1000][1000];//记录地图，这里声明为char类型是应为题目每一行的输入可以看成字符串的形式
int book[1000][1000];//记录所属连通图标记
int a[100000];//记录连通图对应的连通度
int next_step[][2] = { {0, 1}, {1, 0}, {0, -1}, {-1, 0} };//记录行走的步数

int main()
{
	int m, n, strx, stry, tx, ty, head, tail, sum；//sum用于记录当前连通图的连通度
	int	cnt = 1;/*第几个连通图*/

	cin >> n >> m;
	for (int i = 0; i < n; i++)
		cin >> map[i];
	
	for (int i = 0; i < n; i++)
		for(int j = 0; j < n; j++)
			if(book[i][j] == 0)
			{
				book[i][j] = cnt; //连通图初始化 这里很重要，每走到一个新的地点，
				                  //先判断其是否属于某个连通图，如果不是一定要初始化
				strx = i; stry = j;
				head = tail = 0;
				que[tail].x = strx;
				que[tail].y = stry;
				tail++; 
				sum = 1;//连通度初始化

				while (head < tail)
				{
					for (int i = 0; i < 4; i++)//分别枚举判断四个方向
					{
						tx = que[head].x + next_step[i][0];
						ty = que[head].y + next_step[i][1];
						if (tx >= 0 && tx < n && ty >= 0 && ty < n && book[tx][ty] == 0 && map[que[head].x][que[head].y] != map[tx][ty])
						//这里一定要注意条件判断顺序，首先要判断数组是否越界，
						{
							book[tx][ty] = cnt;//标记属于第几个连通图
							que[tail].x = tx;
							que[tail].y = ty;
							sum++;
							tail++;
						}
					}
					head++;
				}
				a[cnt++] = sum;//记录对应的连通度
			}

	for (int i = 0; i < m; i++)
	{
		cin >> strx >> stry;
		cout << a[book[strx - 1][stry - 1]] << endl;
	}

	return 0;
}