洛谷-P1141 01迷宫
题目描述
有一个仅由数字0与1组成的\(n \times 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。
C++代码
#include <iostream>
#include <queue>
using namespace std;
#define MAXN 100000
struct node {
int x;
int y;
}p[MAXN];
int n,m,flag[1001][1001],ans[MAXN+1];
int dir[][4]={{-1,0},{0,1},{1,0},{0,-1}};
char a[1001][1001];
queue<node> q;
int bfs(node u, int k) {
int i,j,sum=0;
node v;
q.push(u);
while(!q.empty()) {
u=q.front(),q.pop();
if(flag[u.x][u.y])
continue;
for(i=0;i<4;++i) {
v.x=u.x+dir[i][0];
v.y=u.y+dir[i][1];
if(v.x<1||v.x>n||v.y<1||v.y>n)
continue;
if(a[u.x][u.y]!=a[v.x][v.y])
q.push(v);
}
flag[u.x][u.y]=k;
++sum;
}
return sum;
}
int main() {
int i,j,k=1;
cin>>n>>m;
for(i=1;i<=n;++i)
for(j=1;j<=n;++j)
cin>>a[i][j];
for(i=0;i<m;++i)
cin>>p[i].x>>p[i].y;
for(i=0;i<m;++i) {
if(!flag[p[i].x][p[i].y]) {
ans[k]=bfs(p[i],k);
++k;
}
cout<<ans[flag[p[i].x][p[i].y]]<<endl;
}
return 0;
}