1091. Acute Stroke (30)--BFS数多少块儿

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <cmath>

#include <string>
#include <cstring>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <stack>

using namespace std;

struct node
{
int x,y,z;
};
int N,M,L,T;
int g[2000][200][100]={0};
int book[2000][200][100]={0};
int X[6]={1,-1,0,0,0,0};
int Y[6]={0,0,1,-1,0,0};
int Z[6]={0,0,0,0,1,-1};

bool judge(node a)
{
int x=a.x;
int y=a.y;
int z=a.z;

if( (x<0||y<0||z<0)||(x>=M||y>=N||z>=L) )//出界
return 0;
if(g[x][y][z]==0)//点是0
return 0;
if(book[x][y][z]==1)//已被访问过
return 0;

return 1;
}

int bfs(int x,int y,int z)
{
int cou=0;

book[x][y][z]=1;//原点先标记
cou++;

node tnode;//搜索的元素
tnode.x=x;  tnode.y=y;  tnode.z=z;
queue<node> q;
q.push(tnode);
while(!q.empty())
{
for(int i=0;i<6;i++)
{
int nx=q.front().x+X[i];
int ny=q.front().y+Y[i];
int nz=q.front().z+Z[i];
node tnode2;//周围的node
tnode2.x=nx;  tnode2.y=ny;  tnode2.z=nz;

if(judge(tnode2)==1)
{
q.push(tnode2);
book[nx][ny][nz]=1;
cou++;
}
}

q.pop();
}

return cou;
}
int main()
{
// freopen("in.txt","r",stdin);

scanf("%d %d %d %d",&M,&N,&L,&T);
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 ans=0;
for(int k=0;k<L;k++)
{
for(int i=0;i<M;i++)
{
for(int j=0;j<N;j++)
{
if(g[i][j][k]==1&&book[i][j][k]==0)
{
int cou=bfs(i,j,k);//对每个点进行搜索
if(cou>=T)
{
ans+=cou;
}

}
}
}
}

printf("%d",ans);

return 0;
}