【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;
}