洛谷P1141： 01迷宫（bfs）

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。

解题思路：

遍历字符串，找到周围能到达的字符，这些字符能够移动的格数是相同的。

#include <stdio.h>
#define N 1020
char a[N][N];
int book[N][N],ans[N][N];
int n;
struct data{
	int x;
	int y;
}s[N*N];
int next[4][2] = {
	1,0, -1,0, 0,1, 0,-1
};
void bfs(int x,int y)
{
	int head=0,tail=1,tx,ty,i,j;
	book[x][y] = 1;
	s[head].x = x;
	s[head].y = y;
	while(head < tail)
	{
		for(i=0; i<4; i++)
		{
			tx = s[head].x + next[i][0];
			ty = s[head].y + next[i][1];
			if(tx>0 && tx<=n && ty>0 && ty<=n && book[tx][ty]==0 && a[tx][ty]!=a[s[head].x][s[head].y])
			{
				s[tail].x = tx;
				s[tail].y = ty;
				book[tx][ty] = 1;
				tail++; 
			}
		}
		head++;
	}
	for(i=0; i<tail; i++)
		ans[s[i].x][s[i].y] = tail-1;
}
int main()
{
	int m,i,j,x,y;
	scanf("%d%d", &n, &m);
	for(i=1; i<=n; i++)
		scanf("%s", a[i]+1);
	while(m--)
	{
		scanf("%d%d", &x, &y);
		if(!ans[x][y])
			bfs(x,y);
		printf("%d\n", ans[x][y]+1);
	}
	return 0;
}