PAT (Advanced Level) Practice - 1091 Acute Stroke（30 分）

最新推荐文章于 2020-11-24 16:48:24 发布

原创最新推荐文章于 2020-11-24 16:48:24 发布 · 350 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#PAT (Advanced Level) Practice #1091 Acute Stroke（30 分） #三维

PTA 专栏收录该内容

173 篇文章

订阅专栏

博客围绕三维数组中肿瘤块大小计算问题展开。给定三维数组，0 代表正常，1 代表肿瘤，肿瘤块大小需大于等于 st 才算作肿瘤。采用三维广度优先搜索解题，用 X、Y、Z 数组判断方向，vis 数组标记访问结点，judge 函数判断边界和结点状态。

题目链接：点击打开链接

题目大意：给定一个三维数组，0表示正常1表示有肿瘤，肿瘤块的大小大于等于 st 才算作是肿瘤，让计算所有满足肿瘤块的大小。

解题思路：三维的广度优先搜索。X、Y、Z三个数组判断方向，对每一个点广度优先累计肿瘤块的大小，如果大于等于 st 就把结果累加。用 vis 数组标记当前的点有没有被访问过，被访问过的结点是不会再访问的。judge 判断是否超过了边界，或者是否当前结点为 0 不是肿瘤。

AC 代码

#include<bits/stdc++.h>
#include<cmath>

#define mem(a,b) memset(a,b,sizeof a)
#define ssclr(ss) ss.clear(), ss.str("")
#define INF 0x3f3f3f3f
#define MOD 1000000007

using namespace std;

typedef long long ll;

const int X[6]={1,0,0,-1,0,0};
const int Y[6]={0,1,0,0,-1,0};
const int Z[6]={0,0,1,0,0,-1};

int m,n,l,st;
int g[1300][130][70], vis[1300][130][70];

struct node
{
    int x,y,z;
    node(){}
    node(int x,int y,int z):x(x),y(y),z(z){}
};

int jde(int x,int y,int z)
{
    if(x<0 || y<0 || z<0 || x>=m || y>=n || z>=l) return 0;
    if(vis[x][y][z]==1 || g[x][y][z]==0) return 0;
    return 1;
}

int bfs(int x,int y,int z)
{
    int cnt=0,dx,dy,dz;
    queue<node> q;
    q.push(node(x,y,z));
    vis[x][y][z]=1;

    node tp;
    while(!q.empty())
    {
        tp=q.front(); q.pop();
        cnt++;
        for(int i=0;i<6;i++)
        {
            dx=X[i]+tp.x;
            dy=Y[i]+tp.y;
            dz=Z[i]+tp.z;
            if(jde(dx,dy,dz))
                vis[dx][dy][dz]=1, q.push(node(dx,dy,dz));
        }
    }

    return cnt>=st?cnt:0;
}

int main()
{
    int rs=0;
    scanf("%d%d%d%d",&m,&n,&l,&st);
    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]);

    for(int k=0;k<l;k++)
        for(int i=0;i<m;i++)
            for(int j=0;j<n;j++)
                if(!vis[i][j][k] && g[i][j][k]==1)
                    rs+=bfs(i,j,k);

    printf("%d\n",rs);

    return 0;
}