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[61][130][1300];
int area=0;
int dx[6][3]={{1,0,0},{-1,0,0},{0,1,0},{0,-1,0},{0,0,1},{0,0,-1}};//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<0||y<0||z<0)return 0;
    if(x>=l||y>=n||z>=m)return 0;
    return 1;
}

void bfs(int x,int y,int z){
    queue<p> qu;
    int sum=1; 
    
    k.x=x;k.y=y;k.z=z;
    a[k.x][k.y][k.z]=0;//开始的点别忘了归0 
    qu.push(k);
    
    while(!qu.empty()){
        k=qu.front();
        qu.pop();
        for(int i=0;i<6;i++){
            o.x=k.x+dx[i][0];
            o.y=k.y+dx[i][1];
            o.z=k.z+dx[i][2];
            if(check(o.x,o.y,o.z)&&a[o.x][o.y][o.z]){
                a[o.x][o.y][o.z]=0;//访问过了就归0
                sum++;
                qu.push(o);
            }
        }
    }
    if(sum>=t)
        area+=sum;
}
int main(){
    scanf("%d%d%d%d",&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++)
            scanf("%d",&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); 
        }
    }
    printf("%d\n",area);
    return 0;
}