【PAT】1091 Acute Stroke (30 分)

三维搜索，按照6个邻接理论，总会有重合的部分区域，

因此根据搜索顺序，先访问到的就是一个区域的【不能较真】 

#include <bits/stdc++.h>
using namespace std;
int m,n,l,t;
int arr[1300][130][80];//1286*128*60
struct node{
    int x,y,z;
};
//(1,0,0)  (0,1,0)  (0,0,1)
//(-1,0,0) (0,-1,0) (0,0,-1)
int X[6]={1,0,0,-1,0,0};
int Y[6]={0,1,0,0,-1,0};
int Z[6]={0,0,1,0,0,-1};

int visited[1300][130][80];
bool judgeNode(node a){
    //1：越界
    if(a.x<0 || a.x>=m || a.y<0 || a.y>=n || a.z<0 || a.z>=l) return false;
    //2：被访问过
    if(visited[a.x][a.y][a.z]==1) return false;
    //3：此处没有病变
    if(arr[a.x][a.y][a.z]==0)  return false;
    return true;
}
int bfs(node a){
    int cnt=0;
    queue<node> q;
    q.push(a),visited[a.x][a.y][a.z]=1;
    while(!q.empty()){
        node tmp=q.front();
        q.pop();
        cnt++;
        for(int i=0;i<6;i++){//每个点有6个邻接点,但有可能越界或者非法
            node xx=node{tmp.x+X[i],tmp.y+Y[i],tmp.z+Z[i]};
            if(judgeNode(xx)){
                visited[xx.x][xx.y][xx.z]=1;
                q.push(xx);
            }
        }
    }
    if(cnt>=t) return cnt;//该块附近体积大于t，计入肿块
    else       return 0;
}
int main(){
     cin>>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++)
                cin>>arr[j][k][i];

    int ans=0;
    for(int i=0;i<l;i++)
        for(int j=0;j<m;j++)
            for(int k=0;k<n;k++)
                if(arr[j][k][i]==1 && visited[j][k][i]==0)//排除bfs的根结点非法
                    ans+=bfs(node{j,k,i});
    cout<<ans;
    return 0;
}