题目链接
https://pintia.cn/problem-sets/994805046380707840/problems/994805052626026496
思路
想象成三维的图来遍历,DFS和BFS应该都可以,本题采用的是BFS,唯一一个文本有歧义的点在于,共一个切面他想表达的意思是只看相邻的上下一个单位,不是说z坐标相同的所有点都算联通的,好在样例和图示已经排除了第二种情况。
BFS代码
#include<bits/stdc++.h>
using namespace std;
struct node
{
int x,y,z;
};
int a[1287][130][60];
int m,n,l,T;
int sum=0;
void bfs(int x,int y,int z)
{
a[x][y][z]=0;
int V=0;
node t;
t.x=x,t.y=y,t.z=z;
int dx[]={
-1,0,1,0,0,0},dy[]={
0,1,0,-1,0,0},dz[]={
0,0,0,0,1,-1};
queue<node> q;
q.push(t);
while(q.size())
{
auto p=q.front();
q.pop();
V++;
for(int i=0;i<6;i++)
{
node p1;
p1.x=p.x+dx[i],p1.y=p.y+dy[i],p1.z=p.z+dz[i];
if(p1.x<0||p1.x>=m||p1.y<0||p1.y>=n||p1.z<0||p1.z>=l||(!a[p1.x][p1.y][p1.z])) continue;
a[p1.x][p1.y][p1.z]=0;
q.push(p1);
}
}
if(V>=T)
{
sum=sum+V;
}
}
int main()
{
scanf("%d%d%d%d",&m,&n,&l,&T);
for(int k=0;k<l;k++)
{
for(int i=0;i<m;i++)
{
for(int j=0;j<n;j++)
{
scanf("%d",&a[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(a[i][j][k]==1)
{
bfs(i,j,k);
}
}
}
}
cout<<sum<<'\n';
return 0;
}