团体程序设计天梯赛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 <cstdio>
#include <string>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <vector>
#include <set>
#include <stack>
#include <map>
#include <climits>

using namespace std;

#define LL long long

const int INF=0x3f3f3f3f;

int a[65][1300][135];
bool visit[65][1300][135];
int n,m,l,k;
int sum;
int dir[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;
}pre,nt;

void bfs(int z,int x,int y)
{
    queue<node>q;
    pre.z=z,pre.x=x,pre.y=y;
    visit[z][x][y]=1;
    q.push(pre);
    int ans=0;
    while(!q.empty())
    {
        ans++;
        pre=q.front();
        q.pop();
        for(int i=0;i<6;i++)
        {
            int xx=pre.x+dir[i][0];
            int yy=pre.y+dir[i][1];
            int zz=pre.z+dir[i][2];
            if(zz<1||zz>l||xx<1||xx>m||yy<1||yy>n||visit[zz][xx][yy]||!a[zz][xx][yy]) continue;
            nt.x=xx,nt.y=yy,nt.z=zz;
            visit[zz][xx][yy]=1;
            q.push(nt);
        }
    }
    if(ans>=k) sum+=ans;
}

int main()
{
    while(~scanf("%d %d %d %d",&m,&n,&l,&k))
    {
        sum=0;
        for(int i=1; i<=l; i++)
            for(int j=1; j<=m; j++)
                for(int k=1; k<=n; k++)
                    scanf("%d",&a[i][j][k]);
        memset(visit,0,sizeof visit);
        for(int i=1;i<=l;i++)
        {
            for(int j=1;j<=m;j++)
            {
                for(int k=1;k<=n;k++)
                {
                    if(!a[i][j][k]||visit[i][j][k]) continue;
                    bfs(i,j,k);
                }
            }
        }
        printf("%d\n",sum);
    }
    return 0;
}