1091 Acute Stroke (30 分)难度: 一般 / bfs

Posted 2021-12-12 辉小歌

https://pintia.cn/problem-sets/994805342720868352/problems/994805375457411072
很常见的题，就是求每一个连通块的大小。

#include<bits/stdc++.h>
using namespace std;
const int N=150;
const int M=2000;
const int L=65;
int f[L][M][N],n,m,l,t,ans;
bool st[L][M][N];
struct node{int x,y,z;};
void bfs(int x,int y,int z)
{
    int dx[6]={-1,0,0,1,0,0};
    int dy[6]={0,-1,1,0,0,0};
    int dz[6]={0,0,0,0,1,-1};
    int cnt=1;
    st[z][x][y]=1;
    queue<node>q; q.push({x,y,z});
    while(q.size())
    {
        auto t=q.front(); q.pop();
        x=t.x,y=t.y,z=t.z;
        for(int i=0;i<6;i++)
        {
            int tempx=x+dx[i];
            int tempy=y+dy[i];
            int tempz=z+dz[i];
            if(tempx<0||tempx>=m||tempy<0||tempy>=n||tempz<0||tempz>=l) continue;
            if(st[tempz][tempx][tempy]) continue;
            if(!f[tempz][tempx][tempy]) continue;
            q.push({tempx,tempy,tempz});
            st[tempz][tempx][tempy]=1,cnt++;
        }
    }
    if(cnt>=t) ans+=cnt;
}
int main(void)
{
    cin>>m>>n>>l>>t;
    for(int i=0;i<l;i++)
        for(int j=0;j<m;j++)
            for(int z=0;z<n;z++)
                cin>>f[i][j][z];
    for(int i=0;i<l;i++)
        for(int j=0;j<m;j++)
            for(int z=0;z<n;z++)
                if(!st[i][j][z]&&f[i][j][z]) bfs(j,z,i);
    cout<<ans;
    return 0;
}