天梯赛-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

#include<iostream>
#include<cstdio>
using namespace std;
int mp[1300][130][61];//定义三维数组
bool visited[1300][130][61];
int ans;//累加像素的体积
typedef struct Node{
    int x;
    int y;
    int z;
}Node;
Node que[1200000];
int sum;//统计体积大小
void bfs(int j,int k,int i,int m,int n,int l){
    int next[6][3] = {{1,0,0},{-1,0,0},{0,1,0},{0,-1,0},{0,0,1},{0,0,-1}};//定义立方体的六个方向
    int head = 1,tail = 1;//设置队头队尾
    //入队操作
    que[tail].x = j;
    que[tail].y = k;
    que[tail].z = i;
    visited[j][k][i] = true;
    tail++;
    sum++;//入队体积加1
    while(head<tail){
        for(int i = 0;i<6;i++){
            int tx = que[head].x + next[i][0];
            int ty = que[head].y + next[i][1];
            int tz = que[head].z + next[i][2];
            if(tx<0||ty<0||tz<0||tx>=m||ty>=n||tz>=l)//判断是否越界，l和1傻傻分不清。。。。。。
                continue;
            if(!visited[tx][ty][tz]&&mp[tx][ty][tz]){//如果当前有关系并且该结点没有被访问过
                visited[tx][ty][tz] = true;
                que[tail].x = tx;
                que[tail].y = ty;
                que[tail].z = tz;
                tail++;
                sum++;
            }
        }
        head++;
    }
    return;
}
int main(){
    int m,n,l,t;
    scanf("%d%d%d%d",&m,&n,&l,&t);
    for(int i = 0;i<l;i++)
        for(int j = 0;j<m;j++)
            for(int k = 0;k<n;k++)
                scanf("%d",&mp[j][k][i]);
    int sm = 0;
    for(int i = 0;i<l;i++)
        for(int j = 0;j<m;j++)
            for(int k = 0;k<n;k++)
                if(mp[j][k][i]){
                    sum = 0;
                    bfs(j,k,i,m,n,l);
                    sum>=t?sm += sum:sm += 0;//进行判断
                }
    printf("%d\n",sm);
    return 0;
}