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

题目：肿瘤诊断

题意：找三维图中的连通块的面积，即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;}