7-13 肿瘤诊断

题解

这题的思路很明显，dfs染色问题，我先写了一个dfs，但是发现段错误了，也就是深搜的层数太多，出现了堆栈溢出，然后改成了bfs就过了。

代码

#include <bits/stdc++.h>
using namespace std;

int m,n,l,t,color=2;
int g[63][1300][130];
int d[6][3]={-1,0,0, 1,0,0, 0,1,0, 0,-1,0, 0,0,1, 0,0,-1};

bool check(int z,int x,int y) {
    if (z>=0&&z<l&&x>=0&&x<m&&y>=0&&y<n) {
        return true;
    }
    return false;
}

int dfs(int z,int x,int y) {
    g[z][x][y]=color;
    int res=1;
    for (int i=0;i<6;i++) {
        int dz=z+d[i][0];
        int dx=x+d[i][1];
        int dy=y+d[i][2];
        if (check(dz,dx,dy)&&g[dz][dx][dy]==1) {
            res+=dfs(dz,dx,dy);
        }
    }
    return res;
}

struct Node {
    int x,y,z;
    Node(int _x,int _y,int _z):x(_x),y(_y),z(_z) {}
};

int bfs(int z,int x,int y) {
    int res=1;
    g[z][x][y]=color;
    queue<Node> q;
    q.push(Node(x,y,z));
    while (!q.empty()) {
        Node f=q.front();
        q.pop();
        for (int i=0;i<6;i++) {
            int dz=f.z+d[i][0];
            int dx=f.x+d[i][1];
            int dy=f.y+d[i][2];
            if (check(dz,dx,dy)&&g[dz][dx][dy]==1) {
                res++;
                g[dz][dx][dy]=color;
                q.push(Node(dx,dy,dz));
            }
        }
    }
    return res;
}

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",&g[k][i][j]);
            }
        }
    }
    int ans=0;
    for (int k=0;k<l;k++) {
        for (int i=0;i<m;i++) {
            for (int j=0;j<n;j++) {
                if (g[k][i][j]==1) {
                    int res=bfs(k,i,j);
                    if (res>=t) {
                        ans+=res;
                    }
                    color++;
                }
            }
        }
    }
    printf("%d\n",ans);
    return 0;
}