L3-004 肿瘤诊断 dfs bfs

　　

在诊断肿瘤疾病时，计算肿瘤体积是很重要的一环。给定病灶扫描切片中标注出的疑似肿瘤区域，请你计算肿瘤的体积。

输入格式：

输入第一行给出4个正整数：M、N、L、T，其中M和N是每张切片的尺寸（即每张切片是一个M×N的像素矩阵。最大分辨率是1）；L（≤）是切片的张数；T是一个整数阈值（若疑似肿瘤的连通体体积小于T，则该小块忽略不计）。

最后给出L张切片。每张用一个由0和1组成的M×N的矩阵表示，其中1表示疑似肿瘤的像素，0表示正常像素。由于切片厚度可以认为是一个常数，于是我们只要数连通体中1的个数就可以得到体积了。麻烦的是，可能存在多个肿瘤，这时我们只统计那些体积不小于T的。两个像素被认为是“连通的”，如果它们有一个共同的切面，如下图所示，所有6个红色的像素都与蓝色的像素连通。

输出格式：

在一行中输出肿瘤的总体积。

输入样例：

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

dfs一直爆栈   最后两个点一直过不了   段错误  
换成bfs就能过  还是bfs大法好

#include<bits/stdc++.h>
using namespace std;
//input
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define RI(n) scanf("%d",&(n))
#define RII(n,m) scanf("%d%d",&n,&m);
#define RIII(n,m,k) scanf("%d%d%d",&n,&m,&k)
#define RS(s) scanf("%s",s);
#define LL long long
#define REP(i,N)  for(int i=0;i<(N);i++)
#define CLR(A,v)  memset(A,v,sizeof A)
//////////////////////////////////
#define N 1500
int n,m,z,t;
long long sum=;
int cnt=;
int mp[N][][];
int vis[N][][];
bool inmap(int x,int y,int s)
{
    return x>=&&x<=n&&y>=&&y<=m&&s>=&&s<=z;
}
int dx[]={,-,,,,};
int dy[]={,,,-,,};
int dz[]={,,,,,-};
void dfs(int x,int y,int z)
{
    vis[x][y][z]=;

    for(int i=;i<;i++)
    {
        int a=x+dx[i];
        int b=y+dy[i];
        int c=z+dz[i];
        if(!inmap(a,b,c))continue;
        if(mp[a][b][c]&&!vis[a][b][c])
        {
            cnt++;
            dfs(a,b,c);
        }
    }
    return ;
}

int main()
{
   RIII(n,m,z);RI(t);
   rep(k,,z)
   rep(i,,n)
   rep(j,,m)
   RI(mp[i][j][k]);
   rep(k,,z)
   rep(i,,n)
   rep(j,,m)
   if(mp[i][j][k]&&!vis[i][j][k])
    {
        cnt=;
        dfs(i,j,k);
        if(cnt>=t)sum+=cnt;
    }
   cout<<sum;
    return ;
}

bfs：

#include<stdio.h>
#include<queue>
#include<algorithm>
using namespace std;

int a[][][];
int area=;
int dx[][]={{,,},{-,,},{,,},{,-,},{,,},{,,-}};//6个方向
int n,m,l,t;

struct p{
    int x,y,z;
}k,o;

int check(int x,int y,int z){//x,y,z是否合法
    if(x<||y<||z<)return ;
    if(x>=l||y>=n||z>=m)return ;
    return ;
}

void bfs(int x,int y,int z){
    queue<p> qu;
    int sum=; 

    k.x=x;k.y=y;k.z=z;
    a[k.x][k.y][k.z]=;//开始的点别忘了归0
    qu.push(k);

    while(!qu.empty()){
        k=qu.front();
        qu.pop();
        for(int i=;i<;i++){
            o.x=k.x+dx[i][];
            o.y=k.y+dx[i][];
            o.z=k.z+dx[i][];
            if(check(o.x,o.y,o.z)&&a[o.x][o.y][o.z]){
                a[o.x][o.y][o.z]=;//访问过了就归0
                sum++;
                qu.push(o);
            }
        }
    }
    if(sum>=t)
        area+=sum;
}
int main(){
    scanf("%d%d%d%d",&n,&m,&l,&t);
    for(int k=;k<l;k++){
        for(int i=;i<n;i++){
            for(int j=;j<m;j++)
            scanf("%d",&a[k][i][j]);
        }
    }

    for(int k=;k<l;k++){
        for(int i=;i<n;i++){
            for(int j=;j<m;j++)
                if(a[k][i][j])
                    bfs(k,i,j);
        }
    }
    printf("%d\n",area);
    return ;
}