PAT——1091 Acute Stroke 甲级（bfs）

题目

https://pintia.cn/problem-sets/994805342720868352/problems/994805375457411072

AC代码

#include<bits/stdc++.h>
using namespace std;
struct node{
	int x,y,z;
};
int m,n,l,t;
int a[1301][131][61];
bool vis[1301][131][61];
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};
bool judge(int x,int y,int z)
{
	if(x<0||x>=m||y<0||y>=n||z<0||z>=l) return false;
	else if(a[x][y][z]==0||vis[x][y][z]==true) return false;
	return true;
}
int bfs(int x,int y,int z)
{
	node tmp;
	tmp.x=x,tmp.y=y,tmp.z=z;
	vis[x][y][z]=true;
	int cnt=0;
	queue<node> q;
	q.push(tmp);
	while(q.size())
	{
		node tmp1=q.front();
		q.pop();
		cnt++;
		for(int i=0;i<6;i++)
		{
			int tx=tmp1.x+X[i];
			int ty=tmp1.y+Y[i];
			int tz=tmp1.z+Z[i];
			if(judge(tx,ty,tz))
			{
				vis[tx][ty][tz]=true;
				tmp.x=tx,tmp.y=ty,tmp.z=tz;
				q.push(tmp);
			}
		}
	}
	if(cnt>=t) return cnt;
	else return 0;	
}
int main()
{
	scanf("%d %d %d %d",&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++)
				scanf("%d",&a[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(a[j][k][i]==1&&vis[j][k][i]==false)
					ans+=bfs(j,k,i);
	printf("%d",ans);
}

参考

1091. Acute Stroke (30)-PAT甲级真题（广度优先搜索）

PATA 1091 Acute Stroke（30 分）解题报告