L3-004 肿瘤诊断

题目链接

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

思路:刚开用的dfs发生了两个段错误，看别人博客说是递归太多结果炸了。

代码:

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long int ll;
ll n,m,l,t,s=0,st=0;//行 列 层数 大于等于t 
int dp[65][1300][130]={0};
int f[6][3]={{0,0,-1},{0,-1,0},{0,1,0},{0,0,1},{1,0,0},{-1,0,0}};//层 行 列 
void dfs(int z,int x,int y)
{
	dp[z][x][y]=0;
	for(int i=0;i<6;i++){
		int fz=f[i][0]+z;
		int fx=f[i][1]+x;
		int fy=f[i][2]+y;
		if(fz>=1&&fz<=l&&fx>=1&&fx<=n&&fy>=1&&fy<=m&&dp[fz][fx][fy]){// 1为肿瘤
		    st++;
			dfs(fz,fx,fy);
		}
	}
} 
int main()
{
	cin>>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++){//列数 
				cin>>dp[i][j][k]; // 1为肿瘤 
			}
		}
	}
	for(int i=1;i<=l;i++){//层数 
		for(int j=1;j<=n;j++){//行数 
			for(int k=1;k<=m;k++){//列数 
			    if(dp[i][j][k]){
			        st=1;
			        dfs(i,j,k);	
				    if(st>=t){
					    s+=st;
				    }							    	
				}
			}
		}
	}
	cout<<s<<endl;
}