7-11 肿瘤诊断

7-11 肿瘤诊断（25 分）

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

输入格式：

输入第一行给出4个正整数：$M$、$N$、$L$、$T$，其中$M$和$N$是每张切片的尺寸（即每张切片是一个$M\times N$的像素矩阵。最大分辨率是$1286\times 128$）；$L$（$\le 60$）是切片的张数；$T$是一个整数阈值（若疑似肿瘤的连通体体积小于$T$，则该小块忽略不计）。

最后给出$L$张切片。每张用一个由0和1组成的$M\times 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

思路：实际上就是一个三维的搜索，但是使用dfs会导致段错误，应该是递归的次数太多了，所以使用bfs，在递归层数太多的情况下尽量使用bfs。在比赛的时候使用了dfs，失去了5分，但是当时没想到改成bfs。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
#include <set>
#include <map>
#include <cmath>
#include <queue>

using namespace std;
int _x[]= {1,-1,0,0,0,0};
int _y[]= {0,0,1,-1,0,0};
int _z[]= {0,0,0,0,1,-1};
int vis[65][1300][130];
int mat[65][1300][130];
int m,n,h,t;
struct Node
{
    int z,x,y;
};
int bfs(int z,int x,int y)
{
    Node per {z,x,y};
    queue<Node> p;
    p.push(per);
    int res=1;
    vis[z][x][y]=1;
    while(!p.empty())
    {
        Node temp=p.front();
        p.pop();
        int zz=temp.z;
        int xx=temp.x;
        int yy=temp.y;
        for(int i=0; i<6; i++)
        {
            int idz=zz+_z[i];
            int idx=xx+_x[i];
            int idy=yy+_y[i];
            if(idz>=h||idz<0) continue;
            if(idx>=n||idx<0) continue;
            if(idy>=m||idy<0) continue;
            if(mat[idz][idx][idy]==0) continue;
            if(vis[idz][idx][idy]) continue;
            vis[idz][idx][idy]=1;
            res++;
            p.push(Node {idz,idx,idy});
            vis[idz][idx][idy]=1;
        }
    }

    return res;
}
//int dfs(int z,int x,int y)
//{
//    if(vis[z][x][y]) return 0;
//    int ans=1;
//    vis[z][x][y]=1;
//    for(int i=0; i<6; i++)
//    {
//        int idz=z+_z[i];
//        int idx=x+_x[i];
//        int idy=y+_y[i];
//        if(idz>=h||idz<0) continue;
//        if(idx>=n||idx<0) continue;
//        if(idy>=m||idy<0) continue;
//        if(mat[idz][idx][idy]==0) continue;
//        ans+=dfs(idz,idx,idy);
//    }
//    return ans;
//}
int main()
{
    cin>>n>>m>>h>>t;
    for(int z=0; z<h; z++)
    {
        for(int x=0; x<n; x++)
        {
            for(int y=0; y<m; y++)
            {
                cin>>mat[z][x][y];
            }
        }
    }
    memset(vis,0,sizeof(vis));
    int ans=0;
    memset(vis,0,sizeof(vis));
    for(int z=0; z<h; z++)
    {
        for(int x=0; x<n; x++)
        {
            for(int y=0; y<m; y++)
            {
                if(vis[z][x][y]) continue;
                if(mat[z][x][y]==0) continue;
                int temp=bfs(z,x,y);
                if(temp>=t) ans+=temp;
            }
        }
    }
    cout<<ans<<endl;
    return 0;
}