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

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

输入样例：

3 4 5 2
1 1 1 1
1 1 1 1
1 1 1 1
0 0 1 1
0 0 1 1
0 0 1 1
1 0 1 1
0 1 0 0
0 0 0 0
1 0 1 1
0 0 0 0
0 0 0 0
0 0 0 1
0 0 0 1
1 0 0 0

输出样例：

26

分析：这道题目比较简单，就是在一个三维空间内部进行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;
}