用一个三维数组模拟空间直角坐标系,接收输入文件。再逐个点地访问,如果是未入过队的1,那么以它为起点进行广度优先遍历,统计该块中所有未入过队的1的数目。
#include <cstdio>
#include <queue>
using namespace std;
const int maxN = 128;
const int maxM = 1286;
const int maxL = 60;
struct node{
int x, y, z;
};
int M, N, L, T;
int pixel[maxN][maxM][maxL];
bool inq[maxN][maxM][maxL]={}; //记录是否入过队
int X[6] = {0, 0, 0, 0, 1, -1};
int Y[6] = {0, 0, 1, -1, 0, 0};
int Z[6] = {1, -1, 0, 0, 0, 0};
bool judge(node a);
int BFS(int x, int y, int z);
int main(){
scanf("%d%d%d%d", &M, &N, &L, &T);
for(int z=0; z<L; z++){
for(int y=0; y<M; y++){
for(int x=0; x<N; x++){
scanf("%d", &pixel[x][y][z]);
}
}
}
int count_1=0;
for(int z=0; z<L; z++){
for(int y=0; y<M; y++){
for(int x=0; x<N; x++){
if(pixel[x][y][z]==1 && inq[x][y][z]==false){
int ct = BFS(x, y, z);
if(ct>=T) count_1 += ct;
}
}
}
}
printf("%d", count_1);
return 0;
}
//超越边界的,为0的,已经入队的,不必考虑
bool judge(node a){
if(a.x<0 || a.x>=N || a.y<0 || a.y>=M || a.z<0 || a.z>=L) return false;
if(pixel[a.x][a.y][a.z]==0) return false;
if(inq[a.x][a.y][a.z]==true) return false;
return true;
}
//从点x,y,z开始广度优先周游,统计所有未入过队的1
int BFS(int x, int y, int z){
queue<node> q;
node temp_node = {x, y, z};
q.push(temp_node);
inq[x][y][z] = true;
int ct=0;
while(!q.empty()){
node current_node = q.front();
q.pop();
ct++;
for(int i=0; i<6; i++){
temp_node.x = current_node.x + X[i];
temp_node.y = current_node.y + Y[i];
temp_node.z = current_node.z + Z[i];
if(judge(temp_node)){
q.push(temp_node);
inq[temp_node.x][temp_node.y][temp_node.z] = true;
}
}
}
return ct;
}