一个典型的bfs,注意bfs的for循环的continue,不是break.
题目描述:
在诊断肿瘤疾病时,计算肿瘤体积是很重要的一环。给定病灶扫描切片中标注出的疑似肿瘤区域,请你计算肿瘤的体积。
输入格式:
输入第一行给出4个正整数:M、N、L、T,其中M和N是每张切片的尺寸(即每张切片是一个M×N的像素矩阵。最大分辨率是1286×128);L(≤60)是切片的张数;T是一个整数阈值(若疑似肿瘤的连通体体积小于T,则该小块忽略不计)。
最后给出L张切片。每张用一个由0和1组成的M×N的矩阵表示,其中1表示疑似肿瘤的像素,0表示正常像素。由于切片厚度可以认为是一个常数,于是我们只要数连通体中1的个数就可以得到体积了。麻烦的是,可能存在多个肿瘤,这时我们只统计那些体积不小于T的。两个像素被认为是“连通的”,如果它们有一个共同的切面,如下图所示,所有6个红色的像素都与蓝色的像素连通。
输出格式:
在一行中输出肿瘤的总体积。
输入样例:
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
代码:
#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
int G[62][1300][130];
int m,n,l,t;
struct point
{
int x;
int y;
int z;
};
int ans=0;
void bfs(int nx,int ny,int nz)
{
int sum=0;
point temp,nt;
temp.x = nx;
temp.y = ny;
temp.z = nz;
int next[6][3]={{0,0,1},{0,0,-1},{0,-1,0},{0,1,0},{1,0,0},{-1,0,0}};
queue<struct point> q;
q.push(temp);
G[nz][nx][ny]=0;
while(!q.empty())
{
temp = q.front();
q.pop();
sum++;
for(int i=0; i<6; i++)
{
nt.x = temp.x+next[i][0];
nt.y = temp.y+next[i][1];
nt.z = temp.z+next[i][2];
if(nt.x<1 || nt.x >m || nt.y<1 || nt.y>n || nt.z<1 || nt.z>l || !G[nt.z][nt.x][nt.y]) continue;
q.push(nt);
G[nt.z][nt.x][nt.y]=0;
}
}
if(sum>=t)
{
ans+=sum;
}
// cout<<sum<<endl;
}
int main()
{
cin>>m>>n>>l>>t;
for(int i=1; i<=l; i++)
{
for(int j=1; j<=m; j++)
{
for(int k=1; k<=n; k++)
{
scanf("%d",&G[i][j][k]);
}
}
}
for(int i=1; i<=l; i++)
{
for(int j=1; j<=m; j++)
{
for(int k=1; k<=n; k++)
{
if(G[i][j][k])
{
bfs(j,k,i);
}
}
}
}
printf("%d\n",ans);
return 0;
}