题目链接:点击打开链接
题目大意:给定一个三维数组,0表示正常1表示有肿瘤,肿瘤块的大小大于等于 st 才算作是肿瘤,让计算所有满足肿瘤块的大小。
解题思路:三维的广度优先搜索。X、Y、Z三个数组判断方向,对每一个点广度优先累计肿瘤块的大小,如果大于等于 st 就把结果累加。用 vis 数组标记当前的点有没有被访问过,被访问过的结点是不会再访问的。judge 判断是否超过了边界,或者是否当前结点为 0 不是肿瘤。
AC 代码
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;
}