*浙大PAT甲级 1091 广搜

这个题目很难读懂,查了资料才知道求三位数组的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;
}


阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/luowei5513/article/details/52369977
个人分类: 浙大pat
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭