肿瘤诊断(三维BFS)

 题目地址：https://www.patest.cn/contests/gplt/L3-004

肿瘤诊断 

在诊断肿瘤疾病时，计算肿瘤体积是很重要的一环。给定病灶扫描切片中标注出的疑似肿瘤区域，请你计算肿瘤的体积。

输入格式：

输入第一行给出4个正整数：MM、NN、LL、TT，其中MM和NN是每张切片的尺寸（即每张切片是一个M\times NM×N的像素矩阵。最大分辨率是1286\times 1281286×128）；LL（\le 60≤60）是切片的张数；TT是一个整数阈值（若疑似肿瘤的连通体体积小于TT，则该小块忽略不计）。

最后给出LL张切片。每张用一个由0和1组成的M\times NM×N的矩阵表示，其中1表示疑似肿瘤的像素，0表示正常像素。由于切片厚度可以认为是一个常数，于是我们只要数连通体中1的个数就可以得到体积了。麻烦的是，可能存在多个肿瘤，这时我们只统计那些体积不小于TT的。两个像素被认为是“连通的”，如果它们有一个共同的切面，如下图所示，所有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为什么会有两个案例段错误，难道是爆了系统栈？知道的大神能不能告诉一下，谢了。。。

DFS代码：

#include <iostream> #include <cstdio> #include <cmath> #include <queue> #include <stack> #include <map> #include <algorithm> #include <vector> #include <string> #include <cstring> #include <sstream> using namespace std; int pic[80][1300][150]; bool vis[80][1300][150]; int m,n,h,T; int cnt; int di[10]={-1,1,0,0,0,0}; int dj[10]={0,0,-1,1,0,0}; int dk[10]={0,0,0,0,1,-1}; void dfs(int i,int j,int k) { if(i<0||i>=h||j<0||j>=m||k<0||k>=n) return; if(vis[i][j][k]==true) return; if(pic[i][j][k]==0) return; vis[i][j][k]=true; cnt++; for(int a=0; a<6; a++) { int x=i+di[a]; int y=j+dj[a]; int z=k+dk[a]; dfs(x,y,z); } } int main() { memset(vis,false,sizeof(vis)); scanf("%d%d%d%d",&m,&n,&h,&T); for(int i=0;i<h;i++) { for(int j=0;j<m;j++) { for(int k=0;k<n;k++) { scanf("%d",&pic[i][j][k]); } } } int ans=0; for(int i=0;i<h;i++) { for(int j=0;j<m;j++) { for(int k=0;k<n;k++) { if(vis[i][j][k]==false&&pic[i][j][k]==1) { cnt=0; dfs(i,j,k); if(cnt>=T) { ans+=cnt; } } } } } printf("%d\n",ans); return 0; } 

BFS代码：

#include <iostream> #include <cstdio> #include <cmath> #include <queue> #include <stack> #include <map> #include <algorithm> #include <vector> #include <string> #include <cstring> #include <sstream> using namespace std; int pic[80][1300][150]; bool vis[80][1300][150]; int m,n,h,T; int cnt; int di[10]={-1,1,0,0,0,0}; int dj[10]={0,0,-1,1,0,0}; int dk[10]={0,0,0,0,1,-1}; int ans; struct note { int i,j,k; note(int x,int y,int z) { i=x; j=y; k=z; } }; typedef struct note note; queue<note> Q; void bfs(int i,int j,int k) { note tmp(i,j,k); Q.push(tmp); while(!Q.empty()) { note a=Q.front(); Q.pop(); if(a.i>=0&&a.i<h&&a.j>=0&&a.j<m&&a.k>=0&&a.k<n) { if(vis[a.i][a.j][a.k]==false && pic[a.i][a.j][a.k]==1) { vis[a.i][a.j][a.k]=true; cnt++; for(int c=0; c<6; c++) { int x=a.i+di[c]; int y=a.j+dj[c]; int z=a.k+dk[c]; if(x>=0&&y>=0&&z>=0&&x<h&&y<m&&z<n&&vis[x][y][z]==false&&pic[x][y][z]==1) { note temp(x,y,z); Q.push(temp); } } } } } } int main() { memset(vis,false,sizeof(vis)); scanf("%d%d%d%d",&m,&n,&h,&T); for(int i=0;i<h;i++) { for(int j=0;j<m;j++) { for(int k=0;k<n;k++) { scanf("%d",&pic[i][j][k]); } } } for(int i=0;i<h;i++) { for(int j=0;j<m;j++) { for(int k=0;k<n;k++) { if(vis[i][j][k]==false&&pic[i][j][k]==1) { cnt=0; bfs(i,j,k); if(cnt>=T) { ans+=cnt; } } } } } printf("%d\n",ans); return 0; }