PAT甲1091 Acute Stroke（30 分）

#include <cstdio>
#include <stdio.h>
#include <cstring>
#include <vector>
#include <algorithm>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <iostream>
#include <stack>
#include <queue>
using namespace std;


struct point
{
    int x;
    int y;
    int z;
    point(int _x,int _y,int _z)
    {
        x=_x;
        y=_y;
        z=_z;
    }
};
int G[1300][130][70];
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 vis[1300][130][70];
int M,N,L,T;

bool judge(int x,int y,int z)
{
    if(x>=M||y>=N||z>=L||x<0||y<0||z<0)return false;
    if(G[x][y][z]==1&&vis[x][y][z]==false)return true;
    else return false;
}

int BFS(int x,int y,int z)
{
    queue<point> q;
    point st(x,y,z);
    q.push(st);
    vis[x][y][z]=true;
    int num=0;
    while(!q.empty())
    {
        point top=q.front();
        q.pop();
        num++;
        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))
            {
                q.push(point(newx,newy,newz));
                vis[newx][newy][newz]=true;
            }
        }
    }
    return num;
}



int main()
{
    scanf("%d%d%d%d",&M,&N,&L,&T);
    memset(vis,false,sizeof(vis));
    for(int k=0;k<L;k++)
    {
        for(int i=0;i<M;i++)
        {
            for(int j=0;j<N;j++)
            {
                scanf("%d",&G[i][j][k]);
            }
        }
    }
    int total=0;
    for(int i=0;i<M;i++)
    {
        for(int j=0;j<N;j++)
        {
            for(int k=0;k<L;k++)
            {
                if(G[i][j][k]==1&&vis[i][j][k]==false)
                {
                    int num=BFS(i,j,k);
                    if(num>=T)total+=num;
                }
            }
        }
    }
    printf("%d\n",total);
    system("pause");
    return 0;
}