题目链接:https://www.luogu.com.cn/problem/P1141
问题重述:
有一个仅由数字0与1组成的n×n格迷宫,你每次可以移动到与自身不一样的格子上,问你每个连通块有多少格。
思考:
搜索题,找连通块以及记录大小,可以用bfs或者dfs来做,但是要注意dfs要用记忆化搜索,否则会超时。(流下了不会记忆化的泪水QAQ)
两种写法对比:
可以看到dfs更快,代码量更少,但是bfs思考起来更简单,两种方法各有利弊。
bfs写法:
用队列广搜,边搜边用动态数组记录块的信息,搜完遍历将结果赋给动态数组内位置值。
#include <bits/stdc++.h>
using namespace std;
char a[1005][1005];
int b[1005][1005];
int c[1005][1005];
int x2[4]={0,0,1,-1};
int y2[4]={1,-1,0,0};
struct sa{
int x,y;
};
int n,m;
void bfs(int dx,int dy)
{
int cnt=1;
b[dx][dy]=1;
sa tmp;
queue<sa>q;
vector<sa>p;
q.push({dx,dy});
while(!q.empty())
{
tmp=q.front();
p.push_back(tmp);//动态数组存
q.pop();
int x1=tmp.x;
int y1=tmp.y;
for(int i=0;i<4;i++)
{
int x=x1+x2[i];
int y=y1+y2[i];
if(x>=1&&x<=n&&y>=1&&y<=n&&a[x][y]!=a[x1][y1]&&b[x][y]==0)
{
q.push({x,y});
cnt++;
b[x][y]=1;
}
}
}
vector<sa>::iterator it;
for(it=p.begin();it!=p.end();++it)
{
sa pos=*it;
c[pos.x][pos.y]=cnt;
}
}
int main()
{
int dx,dy;
cin>>n>>m;
for(int i=1;i<=n;i++)
cin>>a[i]+1;
while(m--)
{
scanf("%d%d",&dx,&dy);
if(b[dx][dy]==0)
{
bfs(dx,dy);
printf("%d\n",c[dx][dy]);
}
else
printf("%d\n",c[dx][dy]);
}
return 0;
}
dfs写法:
用f数组标记连通块个数,cnt记录是第几种连通块,vis记录是否访问过该位置。
注意ans数组一定要大!大!大!(流下了re的泪水)。
1000*1000开1e6合适。
#include <bits/stdc++.h>
using namespace std;
const int N=1100;
char a[N][N];
int vis[N][N],f[N][N];
int x2[5]={0,0,0,1,-1};
int y2[5]={0,-1,1,0,0};
int ans[1000005],n,m,cnt,p;
void dfs(int x,int y)
{
f[x][y]=cnt;
ans[f[x][y]]++;
vis[x][y]=1;
for(int i=1;i<=4;i++)
{
int x3=x+x2[i];
int y3=y+y2[i];
if(a[x3][y3]!=a[x][y]&&x3>=1&&x3<=n&&y3>=1&&y3<=n&&vis[x3][y3]==0)
{
vis[x3][y3]=1;
dfs(x3,y3);
}
}
}
int main()
{
//std::ios::sync_with_stdio(0);
//freopen("2.in","r",stdin);
//freopen("out.txt","w",stdout);
cin>>n>>m;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
cin>>a[i][j];
}
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
if(!vis[i][j])
{
cnt++;//若未访问过则是新的连通块,cnt++
dfs(i,j);
}
}
}
int x,y;
for(int i=1;i<=m;i++)
{
scanf("%d %d",&x,&y);
printf("%d\n",ans[f[x][y]]);
}
return 0;
}
最近在复习搜索,发现啥也不会了T A T