您的位置:首页 > 其它

*浙大PAT甲级 1091 广搜

2016-08-30 15:48 381 查看
这个题目很难读懂,查了资料才知道求三位数组的BFS,对于每个出现的1,寻找其连接的1,统计个数如果大于等于T,则将其加入。

AC代码:

#include<iostream>
#include<map>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
#include<list>
#include<set>
#include<stack>
#include<cmath>
#include<vector>
#define ll long long
#define inf 999999999
using namespace std;
int a[62][1288][130];
int n,m,l,t;
int ans=0;
int dx[6]={1,-1,0,0,0,0};
int dy[6]={0,0,-1,1,0,0};
int dz[6]={0,0,0,0,1,-1};
struct node
{
int x,y,z;
};
void bfs(int x,int y,int z)
{
int num=1;
node tmp;
tmp.x=x;
tmp.y=y;
tmp.z=z;
queue<node> q;
a[x][y][z]=0;
q.push(tmp);
while(!q.empty())
{
node rr=q.front();
q.pop();
for(int i=0;i<6;i++)
{
int xx=rr.x+dx[i];
int yy=rr.y+dy[i];
int zz=rr.z+dz[i];
if(xx<l&&xx>=0&&yy<n&&yy>=0&&zz<m&&zz>=0&&a[xx][yy][zz]==1)
{
a[xx][yy][zz]=0;
node tt;
tt.x=xx;
tt.y=yy;
tt.z=zz;
q.push(tt);
num++;
}
}

}
if(num>=t)
{
ans+=num;
}

}
int main()
{

scanf("%d %d %d %d",&n,&m,&l,&t);
for(int i=0;i<l;i++)
{
for(int j=0;j<n;j++)
{
for(int k=0;k<m;k++)
{
scanf("%d",&a[i][j][k]);
}
}
}
for(int i=0;i<l;i++)
{
for(int j=0;j<n;j++)
{
for(int k=0;k<m;k++)
{
if(a[i][j][k]==1)
{
bfs(i,j,k);
}
}
}
}
cout<<ans<<endl;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: