0
点赞
收藏
分享

微信扫一扫

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


题目链接:​​点击打开链接​​

题目大意:给定一个三维数组,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;
}


举报

相关推荐

0 条评论