【CCCC】L3-004 肿瘤诊断 (30分),三维BFS

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!=k,WA3,6,这TM也能过样例?!
			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;
}

已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 技术工厂 设计师:CSDN官方博客 返回首页