PAT A 1091 Acute Stroke (30 分)

一、题目概述

给定L张（层），每张M行N列的核磁共振片，每个像素代表一个单位立方体（即体积为1）。
若两个立方体共面，称为两个立方体互相连接。显然立方体可以和6个其他立方体直接连接，又可以和许多立方体间接连接成为一个区域。
题目要求统计所有中风区域的体积，即中风区域立方体数，且仅区域立方体数达到门限T才算做中风区域。

二、思路

可以把给定数据看做一个无向图，且除了边缘结点每个结点均有六条边，与共面的立方体连接。
BFS遍历，每个中风区域即一个连通分量，若连通分量结点数达到门限，则sum+=cnt。
注意：DFS仍然可解，但是数据量较大时，会因为递归层数过多系统栈溢出，因此只能使用非递归方式解决。

三、代码

#include <iostream>
#include <queue>
using namespace std;
struct index
{
    int x, y, z;
    index operator+(const index a)const
    {
        return {x + a.x, y + a.y, z + a.z};
    }
};
bool judge(index a, int L, int M, int N)
{
    return (a.x >= 0 && a.x < L && a.y >= 0 && a.y < M && a.z >= 0 && a.z < N);
}
int main()
{
    int M, N, L, T, ans = 0;
    index v, pos, dir[6] = {{1, 0, 0}, {-1, 0, 0}, {0, 1, 0}, {0, -1, 0}, {0, 0, 1}, {0, 0, -1}};
    cin >> M >> N >> L >> T;
    bool G[L][M][N];
    for(int i = 0; i < L; ++i)
        for(int j = 0; j < M; ++j)
            for(int k = 0; k < N; ++k)
                cin >> G[i][j][k];
    for(int i = 0; i < L; ++i)
        for(int j = 0; j < M; ++j)
            for(int k = 0, cnt; k < N; ++k)
                if(G[i][j][k])
                {
                    G[i][j][k] = false;
                    cnt = 1;
                    queue<index> q;
                    q.push({i, j, k});
                    while(q.size())
                    {
                        v = q.front();
                        q.pop();
                        for(int l = 0; l < 6; ++l)
                        {
                            pos = v + dir[l];
                            if(judge(pos, L, M, N) && G[pos.x][pos.y][pos.z])
                            {
                                G[pos.x][pos.y][pos.z] = false;
                                q.push(pos);
                                ++cnt;
                            }
                        }
                    }
                    if(cnt >= T)
                        ans += cnt;
                }
    cout << ans;
}