PAT甲级真题1091 急性中风

识别急性中风的重要因素之一是中风区域的体积。

现在将脑部区域看作是一个 M×N×LM×N×L 的立方体，通过仪器检测，我们可以得到每个单位体积的脑部区域是否中风。

已知，当一片连通的中风区域的体积不小于 TT 时，才会对人体产生威胁，属于危险中风区域。

每个单位区域只与其上下左右前后六个方向直接相连的单位区域算作直接连通。

现在，给定脑部整体中风状况分析，请你计算危险中风区域的总体积，即统计所有的体积不小于 TT 的连通中风区域，计算它们的体积和。

输入格式
第一行包含四个整数 M,N,L,TM,N,L,T，其中 M,N,LM,N,L 分别表示脑部区域的长、宽、高。TT 是一个整数阈值，当某一连通中风区域的体积小于 TT 时，直接忽略不予考虑。

接下来由高到底，按层介绍每层的脑部区域的情况。

共 LL 层脑部区域，每层脑部区域都由一个 M×NM×N 的 0101 矩阵表示，其中 00 表示正常区域，11 表示中风区域。

输出格式
输出危险中风区域的总体积。

数据范围
1≤M≤12861≤M≤1286,
1≤N≤1281≤N≤128,
1≤L≤601≤L≤60
输入样例：
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 <iostream>
#include <cstring>
#include <queue>

using namespace std;

const int M = 1286, N = 128, L = 60;

int m, n, l, T;
int g[L][M][N];
struct Node
{
    int x, y, z;
};

int d[][3] = {
    {1, 0, 0},
    {-1, 0, 0},
    {0, 1, 0},
    {0, -1, 0},
    {0, 0, 1},
    {0, 0, -1},
};

int bfs(int x, int y, int z) //高 长 宽
{
    queue<Node> q;
    q.push({x, y, z});
    g[x][y][z] = 0;

    int cnt = 0;
    while (q.size())
    {
        auto t = q.front();
        q.pop();
        cnt ++ ;
        for (int i = 0; i < 6; i ++ )
        {
            int a = t.x + d[i][0], b = t.y + d[i][1],
             c = t.z + d[i][2];
            if (a >= 0 && a < l && b >= 0 && b < m && c >= 0 
            && c < n && g[a][b][c])
            {
                g[a][b][c] = 0;
                q.push({a, b, c});
            }
        }
    }

    return cnt;
}

int main()
{
    scanf("%d%d%d%d", &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 ++ )
                scanf("%d", &g[i][j][k]);

    int res = 0;
    for (int i = 0; i < l; i ++ )
        for (int j = 0; j < m; j ++ )
            for (int k = 0; k < n; k ++ )
                if (g[i][j][k])
                {
                    int t = bfs(i, j, k);
                    if (t >= T) res += t;
                }

    printf("%d\n", res);

    return 0;
}