【C++ 真题】P1141 01迷宫

P1141 01迷宫

题目描述

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

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

输入格式

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

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

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

输出格式

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

输入输出样例 #1

输入 #1

2 2
01
10
1 1
2 2

输出 #1

4
4

说明/提示

对于样例，所有格子互相可达。

• 对于 $20\%$ 的数据，$n\le 10$；
• 对于 $40\%$ 的数据，$n\le 50$；
• 对于 $50\%$ 的数据，$m\le 5$；
• 对于 $60\%$ 的数据，$n,m\le 100$；
• 对于 $100\%$ 的数据，$1\le n\le 1000$，$1\le m\le 100000$。

题解

#include<bits/stdc++.h>
using namespace std;
const int N = 1e3+7;
string g;
int Q[N][N]={0};
int m, n, a, b;
int ans = 1;
int dx[] = {0, -1, 0, 1, 0}, dy[] = {0, 0, 1, 0, -1};	//上右下左
int F[N][N], an[N][N];//是否走过 和 经过点的个数 
struct Node{
	int x, y;
};
queue<Node> q;

void dfs(int x,int y){
	for(int i=1;i<=4;++i){
		int nx = x + dx[i], ny = y + dy[i];	 
		if(nx < 0 || nx >= n || ny < 0 || ny >= n || F[nx][ny] == 1) continue;
		if((Q[x][y] == 0&&Q[nx][ny]== 1) || (Q[x][y]==1&&Q[nx][ny]==0)) {
			ans++;
			F[nx][ny] = 1; 
			q.push({nx, ny});
			dfs(nx,ny);
		}
	}
	return ;
}

int main(){
	cin>>n>>m;
	for(int i=1;i<=n;++i){
		cin>>g;	
		for(int j=0;j<n;++j){
			Q[i-1][j] = g[j] - '0'; 
			F[i-1][j] = 0;
			an[i-1][j] = 0;
		}
	}
	for(int i=0;i<n;++i){
		for(int j=0;j<n;++j){
			ans = 1;
			if(an[i][j] != 0) continue;
			q.push({i, j});
			F[i][j] = 1; 
			dfs(i, j);
			if(ans > n*n) ans = n*n;
			while(!q.empty()){
				an[q.front().x][q.front().y] = ans;
				q.pop();
			}
		}
	} 
	
	for(int i=1;i<=m;++i){
		cin>>a>>b;
		cout<<an[a-1][b-1]<<endl;
	}
	
	return 0;	
}