PAT甲级 1091 Acute Stroke 三维BFS

Solution：

题目大意：
给出一个三维的立方体，这个立方体可以看成由连续的点组成，整个立方体的体积为m x n x l，其中l为立方体的高。其中有些点的值为1，其余的点都为0。再给出一个常数t，我们要求出所有值为1的块（这些块的体积还要大于t，否则不算）的总体积。每个点的体积为1，所以我们取六个方向，然后找到一个值为1的点，不断bfs直到所有点都被访问。把所有块的体积加起来就可以了。

代码如下：

//三维bfs
#include<iostream>
#include<stdio.h>
#include<math.h>
#include<cstring>
#include<queue>
using namespace std;

int mp[1300][150][65];//记录给出的三维图
int visit[1300][150][65];
int m,n,l,t;//t为阈值
int dir[6][3]={{0,0,1},{0,0,-1},{-1,0,0},{1,0,0},{0,-1,0},{0,1,0}};

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

bool judge(int x,int y,int z){//判断是否越界
    if(x>=0&&x<m&&y>=0&&y<n&&z>=0&&z<l){
        return true;
    }
    return false;
}

int bfs(int x,int y,int z){
    int ans=0;
    queue<node> q;
    visit[x][y][z]=1;
    node temp;
    temp.x=x,temp.y=y,temp.z=z,temp.key=mp[x][y][z];
    q.push(temp);
    while(!q.empty()){
        temp=q.front();
        q.pop();
        ans++;
        for(int i=0;i<6;i++){
            node next;
            next.x=temp.x+dir[i][0];;
            next.y=temp.y+dir[i][1];
            next.z=temp.z+dir[i][2];
            if(judge(next.x,next.y,next.z)){
                next.key=mp[next.x][next.y][next.z];
                if(visit[next.x][next.y][next.z]==0&&next.key==1){
                    q.push(next);
                    visit[next.x][next.y][next.z]=1;
                }
            }
        }
    }
    return ans;
}

int main(){
    int total=0;
    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",&mp[j][k][i]);
            }
        }
    }

    for(int i=0;i<l;i++){
        for(int j=0;j<m;j++){
            for(int k=0;k<n;k++){
                if(mp[j][k][i]==1&&visit[j][k][i]==0){
                    int temp=bfs(j,k,i);
                    if(temp>=t){
                        total+=temp;
                    }
                }
            }
        }
    }

    printf("%d",total);
    return 0;
}