洛谷 普及组 P1141 01迷宫

题目描述

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

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

输入格式

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

下面n行，每行n个字符，字符只可能是0或者1，字符之间没有空格。

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

输出格式

m行，对于每个询问输出相应答案。

数据范围

所有格子互相可达。

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

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

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

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

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

输入输出样例

  输入

2 2
01
10
1 1
2 2

  输出

4
4

思路 

深搜。明确输入一个坐标后，输出它能到达的格子数。可以采用连通块的想法完成这道题。

▶编写solve函数。从某个点（看作老大）开始，找出他能到达的相邻的格子（小弟），从这些格子继续搜索，找出他们能到达的相邻格子（还是小弟），循环下去，直到无路可走。所有这些小弟都可看作是老大的势力，记录老大的势力（代码中用w[ ]数组），即此连通块的格子数目

▶为实现输入一个坐标后能找到他老大的势力，还需要一个数组（代码中用root[ ]数组）来存储每个人的所属领地（cnt），领地编号从1开始，老大和他所有小弟为同一个编号。

#include<iostream>
#include<string.h>
using namespace std;
int dir[4][2] = { {0,1}, {-1,0}, {1,0},{0,-1} }; //方向数组，分别往下，左，右，上 
char map[1001][1001]; 
bool vis[1001][1001]; //访问数组，记录哪些点访问过 
int root[1001][1001], w[1000001];  
int cnt=0; //连通块个数
int n, m, tx[100001], ty[100001]; //tx和ty数组是临时变量，存放我们要查找的坐标 
void solve(int x, int y)
{
	if(!vis[x][y]){
		vis[x][y] = true;
		cnt++; 	//某个老大开辟新领地 
		root[x][y] = cnt;
		w[cnt]++;
	}
	for(int k = 0; k < 4; k++){
		int nx = x + dir[k][1], ny = y + dir[k][0]; 
		if(nx>=1 && nx<=n && ny>=1 && ny<=n && !vis[nx][ny] && map[x][y]!=map[nx][ny]){//下一格可走
			vis[nx][ny] = true;
			root[nx][ny] = cnt; //标记领地编号 
			w[cnt]++; 	    //增长老大势力
			solve(nx, ny);
		}
	}
}
int main()
{
	memset(w, 0, sizeof(w)); //初始化势力范围 
	cin >> n >> m;
	for(int i = 1; i <= n; i++)
		for(int j = 1; j <= n; j++)
			cin >> map[i][j]; //生成地图 
	for(int i = 1; i <= n; i++) 
		for(int j = 1; j <= n; j++)
			solve(i, j);      //找小弟 
	for(int i = 1; i <= m; i++) cin >> tx[i] >> ty[i]; 
	for(int i = 1; i <= m; i++) cout << w[root[tx[i]][ty[i]]] << endl; 
	return 0;
}