PAT L3-004 肿瘤诊断(三维广搜)

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

输入格式：

输入第一行给出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

题意

如上

题解

三维广搜，6个方向

代码

 #include<stdio.h>
 #include<queue>
 #include<algorithm>
 using namespace std;

 int a[][][];
 int area=;
 int dx[][]={{,,},{-,,},{,,},{,-,},{,,},{,,-}};//6个方向
 int n,m,l,t;

 struct p{
     int x,y,z;
 }k,o;

 int check(int x,int y,int z){//x,y,z是否合法
     if(x<||y<||z<)return ;
     if(x>=l||y>=n||z>=m)return ;
     return ;
 }

 void bfs(int x,int y,int z){
     queue<p> qu;
     int sum=; 

     k.x=x;k.y=y;k.z=z;
     a[k.x][k.y][k.z]=;//开始的点别忘了归0
     qu.push(k);

     while(!qu.empty()){
         k=qu.front();
         qu.pop();
         for(int i=;i<;i++){
             o.x=k.x+dx[i][];
             o.y=k.y+dx[i][];
             o.z=k.z+dx[i][];
             if(check(o.x,o.y,o.z)&&a[o.x][o.y][o.z]){
                 a[o.x][o.y][o.z]=;//访问过了就归0
                 sum++;
                 qu.push(o);
             }
         }
     }
     if(sum>=t)
         area+=sum;
 }
 int main(){
     scanf("%d%d%d%d",&n,&m,&l,&t);
     for(int k=;k<l;k++){
         for(int i=;i<n;i++){
             for(int j=;j<m;j++)
             scanf("%d",&a[k][i][j]);
         }
     }

     for(int k=;k<l;k++){
         for(int i=;i<n;i++){
             for(int j=;j<m;j++)
                 if(a[k][i][j])
                     bfs(k,i,j);
         }
     }
     printf("%d\n",area);
     return ;
 }