题目描述
有一个仅由数字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
说明/提示
所有格子互相可达。
对于%100%的数据,n≤1000,m≤100000n≤1000,m≤100000。
解题思路:看完题目明显是搜索,但是一开始写了一个bfs直接T了,想了一下发现,其实在同一个连通块,所以直接给连通块标号,每个点记录一下在几号连通块内,记得数组大小开maxn*maxn,考虑每个人一个连通块的情况。
bfs版本,慢一点
#include <iostream>
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e3+;
int dir[][]= {{,},{,-},{-,},{,}};//上下左右
int vis[maxn][maxn],a[maxn][maxn],in[maxn][maxn],sum[maxn*maxn+];
int n,m,ans=;
struct node
{
int x,y;
};
inline void bfs(int x,int y,int k)
{
ans=;
node s;
s.y=y,s.x=x;
vis[x][y]=;
queue <node> Q;
Q.push(s);
while (!Q.empty())
{
node now=Q.front();
Q.pop();
in[now.x][now.y]=k;
for (int i=; i<; i++)
{
node ne;
ne.x=now.x+dir[i][];
ne.y=now.y+dir[i][];
if (ne.x>=&&ne.x<=n&&ne.y>=&&ne.y<=n&&!vis[ne.x][ne.y]&&a[ne.x][ne.y]!=a[now.x][now.y])
{
vis[ne.x][ne.y]=;
ans++;
Q.push(ne);
}
}
}
sum[k]=ans;
}
int main()
{
scanf("%d%d",&n,&m);
for (int i=;i<=n;i++)
{
for (int j=;j<=n;j++)
{
scanf("%1d",&a[i][j]);
}
}
int k=;
for (int i=;i<=n;i++)
{
for (int j=;j<=n;j++)
{
if(in[i][j]==)
{
bfs(i,j,++k);
}
}
}
for (int i=;i<=m;i++)
{
int x,y;
scanf("%d%d",&x,&y);
printf("%d\n",sum[in[x][y]]);
}
return ;
}
dfs版本,简单易写,速度快
#include <iostream>
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e3+;
int dir[][]= {{,},{,-},{-,},{,}};//上下左右
int vis[maxn][maxn],a[maxn][maxn],in[maxn][maxn],sum[maxn*maxn+];
int n,m,ans=;
struct node
{
int x,y;
};
inline void dfs(int x,int y,int k)
{
in[x][y]=k;
vis[x][y]=;
ans++;
for (int i=;i<;i++)
{
int nx=x+dir[i][],ny=y+dir[i][];
if (nx>=&&nx<=n&&ny>=&&ny<=n&&!vis[nx][ny]&&a[nx][ny]!=a[x][y])
dfs(nx,ny,k);
}
sum[k]=ans;
}
int main()
{
scanf("%d%d",&n,&m);
for (int i=;i<=n;i++)
{
for (int j=;j<=n;j++)
{
scanf("%1d",&a[i][j]);
}
}
int k=;
for (int i=;i<=n;i++)
{
for (int j=;j<=n;j++)
{
if(in[i][j]==)
{
dfs(i,j,++k);
ans=;
}
}
}
for (int i=;i<=m;i++)
{
int x,y;
scanf("%d%d",&x,&y);
printf("%d\n",sum[in[x][y]]);
}
return ;
}