2016年天梯赛模拟题集 7-13 肿瘤诊断 (30分)（BFS，连通块）

原题

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

输入格式：

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

输出格式：

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

输入样例：

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

题意

给定一个三维的立体图，求解肿瘤的体积和，并且去除低于体积阈值的体积。每一个肿瘤都是一个连通块（联通的定义积为相邻，由于是三维的，处于中间的点最多有六个点与它相邻，如上述的示意图，当前处于蓝色点，有上下左右前后六个红色点与之相邻），肿瘤的体积也就是连通块内1的个数。

思路

三维广度优先搜索求解连通块
用深度优先搜索在数据较大时会爆栈，最终只能得25分
比较坑的地方是在如何建图
输入的时候是一个一个面进行输入，所以第三维可以理解为高度，这一层循环应该放在最外面，内部的两层循环类似二维矩阵

代码一（BFS 30分）

//7-13 肿瘤诊断 (30分)
#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

struct Point
{
	int x;
	int y;
	int z;
};

bool G[1300][130][66];	//定义成bool(1个字节) , int则是4个字节 
int m , n , l , t , ans , cnt;
//定义六个方向的偏移矩阵 
int dx[8][6] = {{1 , 0 , 0} , {-1 , 0 , 0} , {0 , 1 , 0} , 
					{0 , -1 , 0} , {0 , 0 , 1} , {0 , 0 , -1}};

void bfs(int x , int y , int z)
{
	queue<Point>q;
	q.push({x , y , z});
	G[x][y][z] = 0;
	
	int cnt = 0;
	while(q.size())
	{
		Point front = q.front();
		q.pop();
		
		cnt++;
		for(int i = 0 ; i < 6 ; i++)
		{
			x = front.x + dx[i][0] , y = front.y + dx[i][1] , z = front.z + dx[i][2];
			if(x >= 0 && x < m && y >= 0 && y < n && 
					z >= 0 && z < l && G[x][y][z])
			{
				//这里没有定义vis数组判断是否走过，直接把走过的点置成非肿瘤 
				G[x][y][z] = false;		
				q.push({x , y , z});
			}	
		}		
	}
	
//	cout<<cnt<<" "<<t<<endl;
	if(cnt >= t)
		ans += cnt;
}

int main()
{
	cin>>m>>n>>l>>t;
 	for(int z = 0 ; z < l ; z++)	//高度即第三维放在最外层 
 		for(int i = 0 ; i < m ; i++)
 			for(int j = 0 ; j < n ; j++)
 				cin>>G[i][j][z];
 	
 	for(int z = 0 ; z < l ; z++)
 		for(int i = 0 ; i < m ; i++)
 			for(int j = 0 ; j < n ; j++)
 				if(G[i][j][z])	
 					bfs(i , j , z);
 					
	cout<<ans;
	return 0; 
} 

代码二（DFS，25分）

//7-13 肿瘤诊断 (30分)
#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

struct Point
{
	int x;
	int y;
	int z;
};

bool G[1300][130][66];	//定义成bool(1个字节) , int则是4个字节 
int m , n , l , t , ans , cnt;
//定义六个方向的偏移矩阵 
int dx[8][6] = {{1 , 0 , 0} , {-1 , 0 , 0} , {0 , 1 , 0} , 
					{0 , -1 , 0} , {0 , 0 , 1} , {0 , 0 , -1}};

void dfs(int x , int y , int z)
{
	cnt++;
	for(int i = 0 ; i < 6 ; i++)
	{
		int tx = x + dx[i][0] , ty = y + dx[i][1] , tz = z + dx[i][2];
		if(tx >= 0 && tx < m && ty >= 0 && ty < n && 
				tz >= 0 && tz < l && G[tx][ty][tz])
		{
			G[tx][ty][tz] = 0;
			dfs(tx , ty , tz);
		}	
	}
} 

int main()
{
	cin>>m>>n>>l>>t;
 	for(int z = 0 ; z < l ; z++)	//高度即第三维放在最外层 
 		for(int i = 0 ; i < m ; i++)
 			for(int j = 0 ; j < n ; j++)
 				cin>>G[i][j][z];
 					
 	for(int z = 0 ; z < l ; z++)
 		for(int i = 0 ; i < m ; i++)
 			for(int j = 0 ; j < n ; j++)
 				if(G[i][j][z])	
 				{
 					cnt = 0;
 					G[i][j][z] = 0;
 					dfs(i , j , z);
 					if(cnt >= t)
					 	ans += cnt;	
				}
	
	cout<<ans;
	return 0; 
}