PTA——肿瘤诊断 分数 30 (图论)

---

一、问题描述

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

输入格式：
输入第一行给出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

---

二、基本思路

• 这道题是网格图中的有关连通块的问题，只不过是三维的，因为数据范围不大，我开一个三维数组来存图，在遍历图时多了两个方向，往上一层切片走或往下一层切片走。需要注意的是我们只统计那些体积不小于T的肿瘤。

---

三、代码

#include<bits/stdc++.h>
using namespace std;
struct point{
	int z,x,y;//坐标 
};
int n,m,L,T,ans=0;//ans记录答案 
int g[61][1300][130];//存图 
bool vis[61][1300][130]; //有没有走过 
//方向：右下左上 上层下层  共六个方向  
int dx[]={0,1,0,-1,0,0};
int dy[]={1,0,-1,0,0,0};
int dz[]={0,0,0,0,1,-1};//上层，下层 

queue<point> q;
inline void bfs(int z,int x,int y){
	int res=1;//肿瘤体积 
	q.push({z,x,y});
	vis[z][x][y]=true;
	while(!q.empty()){
		auto t=q.front();//取出队头 
		q.pop();//队头出队 
		for(int i=0;i<6;i++){//往六个方向探索 
			int tz=t.z+dz[i];
			int tx=t.x+dx[i];
			int ty=t.y+dy[i];
			if(tz<1||tx<1||ty<1||tz>L||tx>n||ty>m) continue;//越界 
			if(g[tz][tx][ty]==0) continue;//不是肿瘤 
			if(vis[tz][tx][ty]) continue;//已经探索过 
			res++;
			q.push({tz,tx,ty});//满足要求，入队 
			vis[tz][tx][ty]=true;//入队后，标记已走过 
		}
	}
	//cout<<res<<endl;
	if(res>=T) ans+=res;//只统计那些体积不小于T的
}

int main(){
	cin>>n>>m>>L>>T;
	for(int k=1;k<=L;k++)
	  for(int i=1;i<=n;i++)
	    for(int j=1;j<=m;j++)
		  cin>>g[k][i][j];
	/*
	for(int k=1;k<=T;k++){
	  for(int i=1;i<=n;i++){
	    for(int j=1;j<=m;j++)
	      cout<<g[k][i][j]<<' ';
	    cout<<endl;
	  }
	}
	*/
	for(int k=1;k<=L;k++)
	  for(int i=1;i<=n;i++)
	    for(int j=1;j<=m;j++){
	    	if(g[k][i][j]==1){
	    		bfs(k,i,j);//连通块问题，遍历每个满足要求的点 
			}
		}
	cout<<ans;
	return 0;
}