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 <stdio.h>    
#include <iostream>    
#include <vector>    
#include <string.h>    
#include <algorithm>    
#include <queue>    
#include <set>   
#define INF 0x3f3f3f 
    
using namespace std;   

typedef struct node{
	int x, y, z;
}Node; 

int mat[80][1300][130];
bool flag[80][1300][130] = {0};
int tem = 0;
int M, N, L, T;
int X[6] = {1, 0, 0, -1, 0, 0};
int Y[6] = {0, 1, 0, 0, -1, 0};
int Z[6] = {0, 0, 1, 0, 0, -1};


void bfs(int z, int x, int y)
{
	Node node;
	node.x = x;
	node.y = y;
	node.z = z;
	
	queue<Node> que;
	que.push(node);
	flag[node.z][node.x][node.y] = 1;
	
	while(!que.empty())
	{
		Node n = que.front();
		++tem;
		que.pop();
		
		for(int i = 0; i < 6; ++i)
		{
			Node m;
			int dx = n.x+X[i];
			int dy = n.y+Y[i];
			int dz = n.z+Z[i];
			
			if(dx < 0 || dx >= M || dy < 0 || dy >= N || dz < 0 || dz >= L)
			{
				continue;
			}
			//int m = mat[dz][dx][dy], n = flag[dz][dx][dy];
			if(mat[dz][dx][dy] == 0 || flag[dz][dx][dy] == 1)
			{
				continue;
			}
			
			m.x = dx, m.y = dy, m.z = dz;
			flag[m.z][m.x][m.y] = 1;
			que.push(m);
		}
	}
}

int main()    
{       
  	cin >> M >> N >> L >> T;
  	for(int i = 0; i < L; ++i)
  	{
  		for(int j = 0; j < M; ++j)
  		{
  			for(int k = 0; k < N; ++k)
  			{
  				cin >> mat[i][j][k];
  			}
  		}
  	}

  	int cnt = 0; 
  	for(int i = 0; i < L; ++i)
  	{
  		for(int j = 0; j < M; ++j)
  		{
  			for(int k = 0; k < N; ++k)
  			{
  				if(flag[i][j][k] == 0 && mat[i][j][k])
  				{
  					tem = 0;
  					bfs(i, j, k);
  					
  					if(tem >= T)
  						cnt += tem;
  				}
  			}
  		}
  	}
  	
  	cout << cnt;
        
    return 0;    
}