(L3-004)肿瘤诊断(三维bfs)

原创已于 2024-03-31 19:30:52 修改 · 178 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#搜索

于 2022-03-21 20:50:36 首次发布

搜索与枚举专栏收录该内容

14 篇文章

订阅专栏

本文解析了如何使用BFS算法在给定的三维矩阵中查找大小至少为T的连通块，并通过示例代码展示了如何实现。重点在于理解空间搜索策略和vis数组的使用来标记已访问节点。

题目链接：PTA | 程序设计类实验辅助教学平台

输入样例：

输出样例：

分析：这道题目比较简单，就是在一个三维空间内部进行bfs搜索，我们每次搜索一个尚未搜索过的值为1的连通块，函数返回连通块的大小，每次判断大小与T的关系，如果连通块大小小于T就跳过，如果连通块大小大于等于T就计入答案，最后输出答案即可，我们判断一个点是否搜索过只需要设置一个vis数组即可，当该点入队后就把对应的vis数组设置为true，这样当我们以一个点为起点遍历完整个连通块时整个连通块内的点的vis数组都被设置为true。

下面是代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
#include<map>
#include<cmath>
#include<queue>
using namespace std;
int s[63][1303][133],n,m,l,t,cnt;
bool vis[63][1303][133];
int dx[6]={0,0,0,0,1,-1},dy[6]={0,0,1,-1,0,0},dz[6]={1,-1,0,0,0,0};
struct node{
	int x,y,z;
};
int bfs(int x,int y,int z)
{
	int cnt=1;
	queue<node> q;
	q.push(node{x,y,z});
	vis[z][x][y]=true;
	while(!q.empty())
	{
		int nowx=q.front().x,nowy=q.front().y,nowz=q.front().z;
		q.pop();
		for(int i=0;i<6;i++)
		{
			int nx=nowx+dx[i],ny=nowy+dy[i],nz=nowz+dz[i];
			if(nx>=1&&nx<=n&&ny>=1&&ny<=m&&nz>=1&&nz<=l&&!vis[nz][nx][ny]&&s[nz][nx][ny]!=0)
			{
				q.push(node{nx,ny,nz});
				vis[nz][nx][ny]=true;
				cnt++;
			}
		}
	}
	return cnt;
}
int main()
{
	scanf("%d%d%d%d",&n,&m,&l,&t);
	for(int i=1;i<=l;i++)
	for(int j=1;j<=n;j++)
	for(int k=1;k<=m;k++)
		scanf("%d",&s[i][j][k]);
	int ans=0;
	for(int i=1;i<=l;i++)
	for(int j=1;j<=n;j++)
	for(int k=1;k<=m;k++)
	{
		if(s[i][j][k]&&!vis[i][j][k])
		{
			int tt=bfs(j,k,i);//注意坐标对应 
			if(tt>=t) ans+=tt;
		}
	}
	printf("%d",ans);
	return 0;
}