problem
L3-004 肿瘤诊断 (30分) 在诊断肿瘤疾病时,计算肿瘤体积是很重要的一环。给定病灶扫描切片中标注出的疑似肿瘤区域,请你计算肿瘤的体积。
输入格式: 输入第一行给出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
给出L个N*M的01矩阵(立体),求所有大于T的联通块大小之和。数据范围1e3*1e2*60
solution
很明显就是一个三维搜索,上下左右前后六个面但是三维,DFS层数过多会爆栈,大佬手动加栈,我还是BFS把令a[k][i][j]表示空间第k片(i,j)的位置,爆搜即可。
#include<bits/stdc++.h>
using namespace std
;
int n
, m
, l
, t
, a
[61][129][1287];
int ans
= 0;
struct xyz
{int x
, y
, z
;};
int dx
[6][3] = {{1,0,0},{-1,0,0},{0,1,0},{0,-1,0},{0,0,1},{0,0,-1}};
int check(int x
, int y
, int z
){if(x
<0||y
<0||z
<0||x
>=l
||y
>=n
||z
>=m
)return 0;return 1;}
void bfs(int x
, int y
, int z
){
queue
<xyz
>q
;
q
.push(xyz
{x
,y
,z
});
a
[x
][y
][z
] = 0;
int sum
= 1;
while(q
.size()){
xyz k
= q
.front(); q
.pop();
for(int i
= 0; i
< 6; i
++){
xyz kk
= k
;
kk
.x
+= dx
[i
][0];
kk
.y
+= dx
[i
][1];
kk
.z
+= dx
[i
][2];
if(check(kk
.x
,kk
.y
,kk
.z
) && a
[kk
.x
][kk
.y
][kk
.z
]){
a
[kk
.x
][kk
.y
][kk
.z
] = 0;
sum
++;
q
.push(kk
);
}
}
}
if(sum
>=t
)ans
+= sum
;
}
int main(){
cin
>>n
>>m
>>l
>>t
;
for(int k
= 0; k
< l
; k
++)
for(int i
= 0; i
< n
; i
++)
for(int j
= 0; j
< m
; j
++)
cin
>>a
[k
][i
][j
];
for(int k
= 0; k
< l
; k
++)
for(int i
= 0; i
< n
; i
++)
for(int j
= 0; j
< m
; j
++)
if(a
[k
][i
][j
])bfs(k
,i
,j
);
cout
<<ans
<<endl
;
return 0;
}