洛谷 01迷宫

题目描述

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

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

输入输出格式

输入格式：
第11行为两个正整数n,mn,m。

下面nn行，每行nn个字符，字符只可能是00或者11，字符之间没有空格。

接下来mm行，每行22个用空格分隔的正整数i,ji,j，对应了迷宫中第ii行第jj列的一个格子，询问从这一格开始能移动到多少格。

输出格式：
mm行，对于每个询问输出相应答案。

输入输出样例

输入样例#1： 复制
2 2
01
10
1 1
2 2
输出样例#1： 复制
4
4
说明

所有格子互相可达。

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

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

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

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

对于100%100%的数据，n≤1000,m≤100000n≤1000,m≤100000。
要是直接搜肯定超时，总走10^6个点 所一用联通块 把同一类归到一起 一次搜索可以走到的点答案都一样 类似与并查集

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
using namespace std;
char s[2000][2000];
int a[2000000], flag[1002][1001] = { 0 };//a数组记录第几个联通却之中元素数，flag记录属于第几个联通却
int dx[4] = { 0, 0, -1, 1 }, dy[4] = { 1, -1, 0, 0 };//向四个方向移动
struct so{
	int x, y;
}q[2000000];//记录元素的横纵坐标类似于队列的头元素只不过可以同时记录横纵  总共有10^6个点
int main()
{
	int n, m;
	scanf("%d%d", &n, &m); 
	int i, j;
	for (i = 1; i <= n; i++)	
	for (j = 1; j <= n; j++)
		cin >> s[i][j];
	int d = 0,r=0,f=0,sum;
	for (i = 1; i <= n; i++)
	{
		for (j = 1; j <= n; j++)
		{
			if (flag[i][j] == 0)
			{
				d++;
				r = 1;
				f = 1;
				sum = 1;
				flag[i][j] = d;
				q[f].x = i;
				q[f].y = j;
				while (f <= r)
				{
					for (int k = 0; k < 4; k++)
					{
						int nx = q[f].x + dx[k], ny = q[f].y + dy[k];
						if (nx >= 1 && nx <= n&&ny >= 1 && ny <= n&&flag[nx][ny] == 0 && s[nx][ny] != s[q[f].x][q[f].y])
						{
							r++;
							q[r].x = nx;
							q[r].y = ny;
							flag[nx][ny] = d;
							sum++;
						}

					}
					f++;
				}
				a[d] = sum;
	
			}
			
		}
	}
	while (m--)
	{
		int g, h;
		scanf("%d%d", &g, &h);
		printf("%d\n", a[flag[g][h]]);
	}
	return 0;
}