L3-004 肿瘤诊断 (30 分)（三维BFS / 三维 DFS）

题目链接

输入样例：

3 4 5 2
1 1 1 1
1 1 1 1
1 1 1 1
0 0 1 1
0 0 1 1
0 0 1 1
1 0 1 1
0 1 0 0
0 0 0 0
1 0 1 1
0 0 0 0
0 0 0 0
0 0 0 1
0 0 0 1
1 0 0 0

输出样例：

26

答案：

法一：

三维BFS

#include <iostream>
#include<bits/stdc++.h>
#define ll long long
#define PII pair<int,int>
#define PSS pair<string,string>
#define PSI pair<string,int>
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))
#define x first
#define y second
#define eps 1e-6
const int N = 1e4 + 10;
using namespace std;

struct node{
    int x;
    int y;
    int z;
};

int m,n,l,t;
int mp[65][1333][133];
int step[6][3]={{0,0,1},{0,0,-1},{0,1,0},{0,-1,0},{1,0,0},{-1,0,0}};

int BFS(node dp){
    queue<node>q;
    q.push(dp);
    mp[dp.x][dp.y][dp.z]=0;
    int tot=1;
    while(q.size()){
        node top=q.front();
        q.pop();
        for(int i=0;i<6;i++){
            node p={top.x+step[i][0],top.y+step[i][1],top.z+step[i][2]};
            if(p.x>0&&p.y>0&&p.z>0&&p.x<=l&&p.y<=m&&p.z<=n&&mp[p.x][p.y][p.z]){
                q.push(p);
                mp[p.x][p.y][p.z]=0;
                tot++;
            }
        }
    }
    return tot;
}

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>>mp[i][j][k];
            }
        }
    }
    int res=0;
    int ans=0;
    for(int i=1;i<=l;i++){
        for(int j=1;j<=m;j++){
            for(int k=1;k<=n;k++){
                if(mp[i][j][k]){
                    res=BFS(node{i,j,k});
                    if(res>=t) ans+=res;
                    else ans+=0;
                }
            }
        }
    }
    cout<<ans<<endl;
    return 0;
}

法二：

三维DFS ，但是最后两个测试点会段错误 不推荐DFS

#include <iostream>
#include<bits/stdc++.h>
#define ll long long
#define PII pair<int,int>
#define PSS pair<string,string>
#define PSI pair<string,int>
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))
#define x first
#define y second
#define eps 1e-6
const int N = 1e4 + 10;
using namespace std;

int m,n,l,t;
int mp[65][1333][133];
int step[6][3]={{0,0,1},{0,0,-1},{0,1,0},{0,-1,0},{1,0,0},{-1,0,0}};

int DFS(int x,int y,int z){
    if(!x||!y||!z||x>l||y>m||z>n||!mp[x][y][z]) return 0;
    mp[x][y][z]=0;
    int res=1;
    for(int i=0;i<6;i++){
        res+=DFS(x+step[i][0],y+step[i][1],z+step[i][2]);
    }
    return res;
}

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>>mp[i][j][k];
            }
        }
    }
    int res=0;
    int ans=0;
    for(int i=1;i<=l;i++){
        for(int j=1;j<=m;j++){
            for(int k=1;k<=n;k++){
                res=DFS(i,j,k);
                if(res>=t) ans+=res;
                else ans+=0;
            }
        }
    }
    cout<<ans<<endl;
    return 0;
}