本质还是搜索(三维空间的搜索)。广搜、深搜应该都可以,深搜可能栈溢出,建议广搜。
用三维数组a存储像素的信息,与搜索配套使用的vis数组在此处即a。
结构体定义:
typedef struct{
int a1,a2,a3;
}node;
六个方向:
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};
坐标合法性:
int check(int x,int y,int z)
{
return x>=1&&x<=l&&y>=1&&y<=m&&z>=1&&z<=n;
}
BFS:
void bfs(int x,int y,int z)
{
a[x][y][z]=0;
node cur,nex;
queue<node>q;
q.push({x,y,z});
while(!q.empty())
{
cur=q.front();
q.pop(); sum++;
for(int i=0;i<=5;i++)
{
int xx=cur.a1+dx[i]; int yy=cur.a2+dy[i]; int zz=cur.a3+dz[i];
if(!check(xx,yy,zz)||!a[xx][yy][zz]) continue ;
a[xx][yy][zz]=0;
q.push({xx,yy,zz});
}
}
}
连起来:
#include<bits/stdc++.h>
using namespace std;
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};
int a[70][1300][130];
int m,n,l,t,ans,sum;
int check(int x,int y,int z)
{
return x>=1&&x<=l&&y>=1&&y<=m&&z>=1&&z<=n;
}
typedef struct{
int a1,a2,a3;
}node;
void bfs(int x,int y,int z)
{
a[x][y][z]=0;
node cur,nex;
queue<node>q;
q.push({x,y,z});
while(!q.empty())
{
cur=q.front();
q.pop(); sum++;
for(int i=0;i<=5;i++)
{
int xx=cur.a1+dx[i]; int yy=cur.a2+dy[i]; int zz=cur.a3+dz[i];
if(!check(xx,yy,zz)||!a[xx][yy][zz]) continue ;
a[xx][yy][zz]=0;
q.push({xx,yy,zz});
}
}
}
int main()
{
cin>>m>>n>>l>>t;
for(int i=1;i<=l;i++)
for(int j=1;j<=m;j++)
for(int k=1;k<=n;k++)
cin>>a[i][j][k];
for(int i=1;i<=l;i++)
for(int j=1;j<=m;j++)
for(int k=1;k<=n;k++)
if(a[i][j][k])
{
bfs(i,j,k);
if(sum>=t)
ans+=sum;
sum=0;
}
cout<<ans;
return 0;
}