PTA甲级-1091

广度优先遍历（bfs）

#include <iostream>
#include <stdio.h>
#include <queue>

using namespace std;
int M,N,L,T;
int m[1287][129][61];
bool visit[1287][129][61]={0};
int ans;
struct axe{
    int x;
    int y;
    int z;
    axe(int a,int b,int c){
        x=a;
        y=b;
        z=c;
    }
};
void bfs(){
    for(int i=0;i<L;i++){
        for(int j=0;j<M;j++){
            for(int k=0;k<N;k++){
                if(visit[j][k][i]||m[j][k][i]!=1){
                    continue;
                }
                int cnt=1;
                queue<axe> que;
                axe a = axe(j,k,i);
                que.push(a);
                visit[a.x][a.y][a.z]=1;
                while(!que.empty()){
                    axe temp = que.front();
                    que.pop();
                    if(temp.x+1<M&&m[temp.x+1][temp.y][temp.z]==1&&!visit[temp.x+1][temp.y][temp.z]){
                        cnt++;
                        axe down = axe(temp.x+1,temp.y,temp.z);
                        que.push(down);
                        visit[temp.x+1][temp.y][temp.z]=1;
                    }
                    if(temp.x-1>=0&&m[temp.x-1][temp.y][temp.z]==1&&!visit[temp.x-1][temp.y][temp.z]){
                        cnt++;
                        axe down = axe(temp.x-1,temp.y,temp.z);
                        que.push(down);
                        visit[temp.x-1][temp.y][temp.z]=1;
                    }
                    if(temp.y+1<N&&m[temp.x][temp.y+1][temp.z]==1&&!visit[temp.x][temp.y+1][temp.z]){
                        cnt++;
                        axe down = axe(temp.x,temp.y+1,temp.z);
                        que.push(down);
                        visit[temp.x][temp.y+1][temp.z]=1;
                    }
                    if(temp.y-1>=0&&m[temp.x][temp.y-1][temp.z]==1&&!visit[temp.x][temp.y-1][temp.z]){
                        cnt++;
                        axe down = axe(temp.x,temp.y-1,temp.z);
                        que.push(down);
                        visit[temp.x][temp.y-1][temp.z]=1;
                    }
                    if(temp.z+1<L&&m[temp.x][temp.y][temp.z+1]==1&&!visit[temp.x][temp.y][temp.z+1]){
                        cnt++;
                        axe down = axe(temp.x,temp.y,temp.z+1);
                        que.push(down);
                        visit[temp.x][temp.y][temp.z+1]=1;
                    }
                    if(temp.z-1>=0&&m[temp.x][temp.y][temp.z-1]==1&&!visit[temp.x][temp.y][temp.z-1]){
                        cnt++;
                        axe down = axe(temp.x,temp.y,temp.z-1);
                        que.push(down);
                        visit[temp.x][temp.y][temp.z-1]=1;
                    }
                }
                if(cnt>=T)
                    ans+=cnt;
            }
        }
    }
}
int main()
{
    //freopen("D:/test.txt","r",stdin);
    cin>>M>>N>>L>>T;
    int i;
    for(i=0;i<L;i++){
        for(int j=0;j<M;j++){
            for(int k=0;k<N;k++){
                cin>>m[j][k][i];
            }
        }
    }
    bfs();
    cout<<ans;
    return 0;
}