GPLT L3-004. 肿瘤诊断【三维bfs求连通块】

来源：互联网发布：云计算与虚拟化的区别编辑：程序博客网时间：2024/05/19 13:10

题目：肿瘤诊断

题意：找三维图中的连通块的面积，即1的个数连通块有个要求如果小于给定的范围就不计！

思路：开始时用dfs做，栈溢出了。。。数据大了！

换成bfs寻找连通即可！

代码：

#include <stdio.h>#include <cstring>#include <queue>using namespace std;const int row = 1500;const int col = 200;const int heg = 100;const int dirx[] = {1,-1,0,0,0,0};const int diry[] = {0,0,-1,1,0,0};const int dirz[] = {0,0,0,0,1,-1};int g[heg][row][col];int n,m,l,t,cnt;struct node{    int x,y,z;    node(int tx,int ty,int tz):x(tx),y(ty),z(tz){};};void bfs(int sz,int sx,int sy){    g[sz][sx][sy] = 0;    queue<node>Q;    Q.push(node(sx,sy,sz));    while(!Q.empty()){        node pre = Q.front();Q.pop();        for(int i=0;i<6;i++){            int tx = pre.x+dirx[i] , ty = pre.y+diry[i] , tz = pre.z + dirz[i];            if(tx < 0 || tx >= n || ty < 0 || ty >= m || tz < 0 || tz >= l) continue;            if(g[tz][tx][ty])            {                g[tz][tx][ty] = 0;                Q.push(node(tx,ty,tz));                cnt++;            }        }    }    return;}int main(){    while(~scanf("%d%d%d%d",&n,&m,&l,&t)){        for(int k=0;k<l;k++)            for(int i=0;i<n;i++)                for(int j=0;j<m;j++) scanf("%d",&g[k][i][j]);        int ans = 0;        for(int k=0;k<l;k++)            for(int i=0;i<n;i++)                for(int j=0;j<m;j++){                    if(g[k][i][j]){                        cnt = 1;                        bfs(k,i,j);                        if(cnt >= t) ans += cnt;                    }                }        printf("%d\n",ans);    }    return 0;}

0 0