PAT A1091 Acute Stroke

对于坐标平面的bfs模板题~

#include<bits/stdc++.h>
using namespace std;
const int maxn=;
bool visit[][][]={false};
int adj[][][];
int n,m,l,K;
int X[]={,,,-,,};
int Y[]={,,,,-,};
int Z[]={,,,,,-};
struct node {
    int x,y,z;
};
bool judge (int x,int y,int z) {
    if (x>=n||x<||y>=m||y<||z>=l||z<)
    return false;
    if (adj[x][y][z]==) return false;
    if (visit[x][y][z]==true) return false;
    return true;
}
int bfs (int x,int y,int z) {
    queue<node> q;
    q.push({x,y,z});
    int cnt=;
    visit[x][y][z]=true;
    while (!q.empty()) {
        node now=q.front();
        q.pop();
        cnt++;
        for (int i=;i<;i++) {
            int tx=now.x+X[i];
            int ty=now.y+Y[i];
            int tz=now.z+Z[i];
            if (judge(tx,ty,tz)) {
                q.push({tx,ty,tz});
                visit[tx][ty][tz]=true;
                //cnt++;
            }
        }
    }
    if (cnt>=K) return cnt;
    return ;
}
int main () {
    scanf ("%d %d %d %d",&n,&m,&l,&K);
    for (int i=;i<l;i++)
    for (int j=;j<n;j++)
    for (int k=;k<m;k++)
    scanf ("%d",&adj[j][k][i]);
    int cnt=;
    for (int i=;i<l;i++)
    for (int j=;j<n;j++)
    for (int k=;k<m;k++)
    if (visit[j][k][i]==false&&adj[j][k][i]==) cnt+=bfs(j,k,i);
    printf ("%d\n",cnt);
    return ;
}