PAT天梯赛L3-004 肿瘤诊断

题目链接：点击打开链接

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

输入格式：

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

思路：看到样例就是要广搜。题意中是立体的，所以要用三维数组存储。 还有就是 只看周围的六个方向，不带自己。

AC代码：

#include<iostream>
#include<queue>
#include<algorithm>
#include<stack>
#include<string>
#include<map>
#include<set>
#include<cstring>
using namespace std;
const int MAX = 1010;
const int INF = 0X3f3f3f;
bool arr[1300][130][80];//bool类型 不会超内存
bool vis[1300][130][80];
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};

int n, m, l, t;
struct node{
	int x;
	int y;
	int z;
	node() {}
	node(int _x, int _y, int  _z):x(_x), y(_y), z(_z) {}
};

bool judge(int x, int y, int z) { //判断该点是否需要访问
	if(x < 0 || x >= n || y < 0 || y >= m || z < 0 || z >= l) return false;//出界
	if(arr[x][y][z] == false || vis[x][y][z] == true) return false;//该点为0 或者 被访问过了
	return true;
}

int bfs(int x, int y, int z) {
	int ans = 0;//记录周围1 的个数
	queue<node> q;//开队列
	q.push(node(x, y, z));//起点压入
	vis[x][y][z] = true;//标记访问过
	while(!q.empty()) {//不为空 就一直继续
		node top = q.front();//取出队首元素
		q.pop();//清除队首元素
		++ans;//在这里累加
		for(int i = 0; i < 6; i++) { // 用数组来访问周围的点
			int newx = top.x + X[i];
			int newy = top.y + Y[i];
			int newz = top.z + Z[i];
			if(judge(newx, newy, newz)) { // 判断是否符合
 				q.push(node(newx, newy, newz));//压入
				vis[newx][newy][newz] = true;//标记
			}
		}
	}
	return ans >= t ? ans : 0; // 满足阈值 就返回
}

int main() {
	int i, j, k;
	//////////////////////////////////////       i， j， k    容易弄混 厚度 应该在最外层   也不要打错
	cin >> n >> m >> l >> t;
	for(int i = 0; i < l; i++) {
		for(int j = 0; j < n; j++) {
			for(int k = 0; k < m; k++) {
				scanf("%d", &arr[j][k][i]);
			}
		}
	}
	int sum = 0;
	for(int i = 0; i < l; i++) {
		for(int j = 0; j < n; j++) {
			for(int k = 0; k < m; k++) {
				//这里的条件我刚开始用judge判的，出错了
				if(arr[j][k][i] == true && vis[j][k][i] == false) {
					sum += bfs(j, k, i);
				}
			}
		}
	}
	cout << sum;
	return 0;
}