L3-004 肿瘤诊断 (30分)(简单三维bfs)

题目链接
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;
}

猜你喜欢

转载自blog.csdn.net/Star_Platinum14/article/details/112727003
今日推荐