luogu P1141 01迷宫

https://www.luogu.org/problem/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。

代码

#include<algorithm>
#include<iostream>
#include<string.h>
#include<stdio.h>
#include<queue>
using namespace std;

const int maxn=1010;
int d[8]={0, 1,  0,-1,  1, 0,  -1, 0};// 方位数组 
char g[maxn][maxn];// 存储地图 
int visited[maxn][maxn];// 给搜索过的位置作标记 
int n,m;
int ans[maxn*100];// 存储第t次搜索的结果 

struct point
{
	int x,y;
	point(int a,int b):x(a),y(b){}
};

int borderline(int x,int y)
{
	return (x>0 && x<=n && y>0 && y<=n);
}

int bfs(int x,int y,int number)// 第number次搜索 
{
	int cnt=1;
	queue <point> q;
	q.push(point(x,y));

	while(!q.empty())
	{
		point t=q.front();q.pop();

		for(int i=0; i<8; i+=2)
		{
			int tx=t.x+d[i];
			int ty=t.y+d[i+1];

			if(borderline(tx,ty) && g[t.x][t.y]!=g[tx][ty] && visited[tx][ty]==0)
			{
				q.push(point(tx,ty));
				visited[tx][ty]=number;
				cnt++;
			}
		}
	}

	return cnt;
}

int main()
{
	cin>>n>>m;
	for(int i=1; i<=n; i++)
		cin>>g[i]+1;// 下标从1~n 

	for(int k=1; k<=m; k++)
	{
		int x,y;
		cin>>x>>y;
		if(visited[x][y]==0)// 如果该格子没有被搜索过，就要搜索一遍  
		{
			visited[x][y]=k;
			ans[k]=bfs(x,y,k);
			cout<<ans[k]<<endl;
		}
		else// 被搜索过的话，直接输出结果  
		{
			int t=visited[x][y];
			ans[m]=ans[t];
			cout<<ans[m]<<endl;
		}
	}

	return 0;
}