pat 甲题 1091

题目：https://pintia.cn/problem-sets/994805342720868352/problems/994805375457411072】

 

使用BFS进行遍历和记录1的个数。

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

struct node{
	int x,y,z;
} Node;

int n,m,slice,T; // n*m的矩阵， slice层 
int pixel[1290][130][61];
bool inq[1290][130][61] = {false};
int X[6]{0, 0, 1,-1, 0, 0};
int Y[6]{1,-1, 0, 0, 0, 0};
int Z[6]{0, 0, 0, 0, 1,-1};
 

bool judge(int x, int y, int z){
	if(x<0 || x>=n || y<0 || y>=m || z<0 || z>=slice) return false; //边界
	if(inq[x][y][z] == true || pixel[x][y][z] == 0) return false;
	return true; 
}

int BFS(int x, int y, int z) {
	queue<node>q;
	Node.x = x;
	Node.y = y;
	Node.z = z;
	q.push(Node);
	int nowT = 1;
	inq[x][y][z] = true;
	while(!q.empty()) {
		node top = q.front();
		q.pop();
		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)) {
				Node.x = newX;
				Node.y = newY;
				Node.z = newZ;
				q.push(Node);
				nowT++;
				inq[newX][newY][newZ] = true;
			}
		}
	}
	if(nowT >= T) return nowT;
	else return 0;
	
}
int main() {
	
	scanf("%d%d%d%d", &n, &m, &slice, &T);
	for (int z=0; z<slice;z++){
		for(int x=0; x<n; x++) {
			for(int y=0; y<m; y++) {
				scanf("%d", &pixel[x][y][z]);
			}
		}
	}
	
	int ans = 0;
	
		for (int z=0; z<slice; z++){
		for(int x=0; x<n; x++) {
			for(int y=0; y<m; y++) {
				if(pixel[x][y][z]==1 && inq[x][y][z]==false){
					ans += BFS(x,y,z);
				}
			}
		}
	}
	
	printf("%d\n", ans);
	return 0;
}

 

这种矩阵的遍历使用dfs和bfs都有一个固定的模式，平时多做题，要总结。就和高中记某一类题型的解法一样。

推荐《算法笔记》一书，入门与刷题神书。