【PAT-A1091】Acute Stroke（图的遍历，三维）

【题意】

给出一个三维的图，求其中块的大小大于阈值的连通块的个数。
三维图的给出方式为先给出每层的长宽（M*N）和层数（L）以及阈值（T），再依次给出L层的01矩阵。

【思路】

就是简单的图的遍历DFS和BFS均可，遍历的同时求出满足阈值的连通块数目。

【注意点】

• 这里我开始用的是DFS后来发现这里再在数据量大的时候DFS容易由于递归太深而爆栈，所以以后遇到这种数据量大的题还是最好用BFS保险。
• 不知道为什么这题我的代码用g++会编译错误，但是用clang++就不会， 后面再找找原因。
• 还有就是在写BFS的时候犯了一个错，在寻找相邻结点的时候，没有用top的xyz而是用的BFS函数的ijk导致结果出错。在写代码的时候还是要想清楚原理，而不是凭记忆和模板蒙头写。

【代码】

#include <cstdio>
#include <algorithm>
#include <iostream>
#include <queue>
using namespace std;

struct Node
{
	int x, y, z;
	Node(int x_, int y_, int z_) {
		x = x_;
		y = y_;
		z = z_;
	}
};

const int maxn = 128;
const int maxm = 1286;
const int maxl = 60;


int graph[maxl][maxm][maxn] = { -1 };
bool vis[maxl][maxm][maxn] = { false };
int num = 0;
int m, n, l, t;

// 用DFS应该是递归过深空间复杂度太大，会爆栈。本地测试没有问题
void DFS(int i, int j, int k, int &cnt) {
	vis[i][j][k] = true;
	cnt++;
	int X[6] = {1, -1, 0, 0, 0, 0};
	int Y[6] = {0, 0, 1, -1, 0, 0};
	int Z[6] = {0, 0, 0, 0, 1, -1};
	for (int p = 0; p < 6; p++)
	{
		int z = i + X[p];
		int x = j + Y[p];
		int y = k + Z[p];
		if ((z >= 0 && z < l) && (x >= 0 && x < m) && (y >= 0 && y < n)) {
			if (vis[z][x][y] == false && graph[z][x][y] == 1) {
				DFS(z, x, y, cnt);
			}
		}

	}
}

int BFS(int i, int j, int k) {
	// 因为BFS没有递归，所以用返回值就行
	int cnt = 0; //记录块的大小
	queue<Node> q;
	q.push(Node(j, k, i));
	vis[i][j][k] = true;
	while (!q.empty())
	{
		Node top = q.front();
		q.pop();
		// 处理结点信息出队时再进行
		cnt++;
		int X[6] = { 1, -1, 0, 0, 0, 0 };
		int Y[6] = { 0, 0, 1, -1, 0, 0 };
		int Z[6] = { 0, 0, 0, 0, 1, -1 };
		for (int p = 0; p < 6; p++)
		{
			int z = top.z + Z[p];
			int x = top.x + X[p];
			int y = top.y + Y[p];
			if ((z >= 0 && z < l) && (x >= 0 && x < m) && (y >= 0 && y < n)) {
				if (vis[z][x][y] == false && graph[z][x][y] == 1) {
					q.push(Node(x, y, z));
					vis[z][x][y] = true;
				}
			}

		}
	}

	if (cnt >= t) {
		return cnt;
	}
	else return 0;
}

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", &graph[i][j][k]);
			}
		}
	}

	for (int i = 0; i < l; i++)
	{
		for (int j = 0; j < m; j++)
		{
			for (int k = 0; k < n; k++)
			{
				if (vis[i][j][k] == false && graph[i][j][k] == 1) {
					num += BFS(i, j, k);
				}
			}
		}
	}

	printf("%d\n", num);
	
	return 0;
}