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 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<queue>
using namespace std;
int n,m,l,t,v=0,map[61][129][1287];
int d[6][3]={{1,0,0},{-1,0,0},{0,1,0},{0,-1,0},{0,0,1},{0,0,-1}};
struct Node
{
    int x,y,z;
};
int judge(int x,int y,int z)
{
    if(x<0 || y<0 || z<0) return 0;
    if(x>=l || y>=n || z>=m) return 0;
    return 1;
}
void bfs(int x,int y,int z)
{
    int i,sum=1;
    queue<Node>q;
    map[x][y][z]=0;
    struct Node u;
    u.x=x,u.y=y,u.z=z;
    q.push(u);
    struct Node p;
    while(!q.empty())
    {
        u=q.front(); q.pop();
        for(i=0;i<6;i++)
        {
            p.x=u.x+d[i][0];
            p.y=u.y+d[i][1];
            p.z=u.z+d[i][2];
            if(judge(p.x,p.y,p.z)==1 && map[p.x][p.y][p.z]==1)
            {
                sum++;
                map[p.x][p.y][p.z]=0;
                q.push(p);
            }
        }  
    }
    if(sum>=t) v=v+sum;
}
int main()
{
    cin>>n>>m>>l>>t;
    int i,j,k;
    for(i=0;i<l;i++)
    {
        for(j=0;j<n;j++)
        {
            for(k=0;k<m;k++)
                cin>>map[i][j][k];
        }
    }
    for(i=0;i<l;i++)
    {
        for(j=0;j<n;j++)
        {
            for(k=0;k<m;k++)
            {
                if(map[i][j][k]==1)
                    bfs(i,j,k);
            }
        }
    }
    cout<<v<<endl;
    return 0;
}