L3-004. 肿瘤诊断

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

题目看懂了其实就是搜索的一种变形，以前接触的多的是平面内的搜索，但是这道题加入了“立体”的概念，也就是在多个叠加的平面内搜索。在平面搜搜索的时候我们会给每个坐标点加上上下左右四个方向。那现在是立体搜索，那么就有 6 个方向，即多了上一层的相同横纵坐标点和下一层的相同横纵坐标点。说白了就是平面内的搜索有 x 和 y 两个坐标轴，现在变成了立体的搜索，那就要加上一个坐标轴即 z 轴。原来只有 x， y 坐标轴的时候有 4 个 方向，现在加上 z 轴就有 6 个方向。值的注意的是如果使用 dfs（深度优先搜索）是会段错误的，应该是递归层次太深，程序栈内存不足了。

第一份代码是直接用 dfs 做的，但是最后两个测试点段错误

// <!! dfs 搜索，最后两个测试点段错误，应该是递归层数太深爆栈了 !!> 
#include <iostream>
using namespace std;

int m, n, l, t;
int area[65][1300][130];
bool visit[65][1300][130];
// 一个坐标点的立体周围 6 个方向: z, x, y 
int directions[6][3] = {
  {
  0, -1, 0}, {
  0, 1, 0}, {
  0, 0, -1},
                         {
  0, 0, 1}, {
  1, 0, 0}, {-1, 0, 0}}; 

int dfs(int z, int x, int y) {
    int res = 0;
    if (z < 0 || z >= l || x < 0 || x >= m || y < 0 || y >= n || 
        !area[z][x][y] || visit[z][x][y]) {
            return res;
    }
    res++;
    visit[z][x][y] = true;