团体程序设计天梯赛-L3-004. 肿瘤诊断

L3-004. 肿瘤诊断

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

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

输入格式：

输入第一行给出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个红色的像素都与蓝色的像素连通。

Figure 1

输出格式：

在一行中输出肿瘤的总体积。

输入样例：

3 4 5 21 1 1 11 1 1 11 1 1 10 0 1 10 0 1 10 0 1 11 0 1 10 1 0 00 0 0 01 0 1 10 0 0 00 0 0 00 0 0 10 0 0 11 0 0 0

输出样例：

26

三维搜，注意只要开一个数组，两个会re。

#include <cstdio>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <map>
#include <queue>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
const int N=1e5+10;
int n,m,l,T;
int dir[6][3]= {1,0,0,-1,0,0,0,-1,0,0,1,0,0,0,1,0,0,-1};
int a[66][130][1300];
struct node
{
    int x,y,z;
};
bool check(int x,int y,int z)
{
    if(x>=1 &&x<=l &&y>=1 &&y<=n &&z>=1 &&z<=m &&a[x][y][z]==1)
        return 1;
    else
        return 0;
}
int bfs(int x,int y,int z)
{
    int sum=1;
    node st,ed;
    st.x=x;
    st.y=y;
    st.z=z;
    a[x][y][z]=0;
    queue<node>q;
    q.push(st);
    while(!q.empty())
    {
        st=q.front();
        q.pop();
        for(int i=0; i<6; i++)
        {
            ed.x=st.x+dir[i][0];
            ed.y=st.y+dir[i][1];
            ed.z=st.z+dir[i][2];
            if(check(ed.x,ed.y,ed.z))
            {
                sum++;
                a[ed.x][ed.y][ed.z]=0;
                q.push(ed);
            }
        }
    }
    if(sum>=T)
        return sum;
    else
        return 0;
}
int main()
{
    while(cin>>n>>m>>l>>T)
    {
        int i,j,k;
        for(i=1; i<=l; i++)
            for(j=1; j<=n; j++)
                for(k=1; k<=m; k++)
                    scanf("%d",&a[i][j][k]);
        int ans=0;
        for(i=1; i<=l; i++)
            for(j=1; j<=n; j++)
                for(k=1; k<=m; k++)
                {
                    if(a[i][j][k])
                        ans+=bfs(i,j,k);
                }
        printf("%d\n",ans);
    }
    return 0;
}