【CCCC】L3-004 肿瘤诊断 (30分)，三维BFS

problem

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

给出L个N*M的01矩阵（立体），求所有大于T的联通块大小之和。
数据范围1e3*1e2*60

solution

很明显就是一个三维搜索，上下左右前后六个面
但是三维，DFS层数过多会爆栈，大佬手动加栈，我还是BFS把
令a[k][i][j]表示空间第k片（i,j）的位置，爆搜即可。

#include<bits/stdc++.h>
using namespace std;int n, m, l, t, a[61][129][1287];int ans = 0;
struct xyz{int x, y, z;};
int dx[6][3] = {{1,0,0},{-1,0,0},{0,1,0},{0,-1,0},{0,0,1},{0,0,-1}};
int check(int x, int y, int z){if(x<0||y<0||z<0||x>=l||y>=n||z>=m)return 0;return 1;}
void bfs(int x, int y, int z){queue<xyz>q;  q.push(xyz{x,y,z}); a[x][y][z] = 0;int sum = 1;while(q.size()){xyz k = q.front(); q.pop();for(int i = 0; i < 6; i++){xyz kk = k;//kk!=k,WA3,6,这TM也能过样例？！kk.x += dx[i][0];kk.y += dx[i][1];kk.z += dx[i][2];if(check(kk.x,kk.y,kk.z) && a[kk.x][kk.y][kk.z]){a[kk.x][kk.y][kk.z] = 0;sum++;q.push(kk);}}}if(sum>=t)ans += sum;
}int main(){cin>>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++)cin>>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);cout<<ans<<endl;return 0;
}