FZU 1063 三维扫描

水题。DFS求连通块。

#include<cstdio>
#include<cstring>
#include<cmath>
#include <iomanip>
#include<iostream>
#include<algorithm>
using namespace std;

const int maxn=+;
int f[maxn][maxn][maxn];
int flag[maxn][maxn][maxn];
int L,W,H;
int M;
int Block;

int dir[][]={
    {,,},
    {,,-},
    {,,},
    {,-,},
    {,,},
    {-,,}
};

void read()
{
    scanf("%d",&M);
    for(int i=;i<=L;i++)
        for(int j=;j<=W;j++)
            for(int k=;k<=H;k++)
                scanf("%d",&f[i][j][k]);
}

void init()
{
    memset(flag,,sizeof flag);
    Block=;
}

bool P(int a,int b,int c)
{
    if(a>=&&a<=L)
    {
        if(b>=&&b<=W)
        {
            if(c>=&&c<=H)
            {
                return ;
            }
        }
    }
    return ;
}

void dfs(int x,int y,int z)
{
    flag[x][y][z]=Block;

    for(int i=;i<;i++)
    {
        int new_x=x+dir[i][];
        int new_y=y+dir[i][];
        int new_z=z+dir[i][];

        int R=f[x][y][z];
        int F=f[new_x][new_y][new_z];

        if(P(new_x,new_y,new_z)&&flag[new_x][new_y][new_z]==&&abs(R-F)<=M)
            dfs(new_x,new_y,new_z);
    }
}
void work()
{
    for(int i=;i<=L;i++)
        for(int j=;j<=W;j++)
            for(int k=;k<=H;k++)
            {
                if(flag[i][j][k]==)
                {
                    Block++;
                    dfs(i,j,k);
                }
            }
    printf("%d\n",Block);
}

int main()
{
    while(~scanf("%d%d%d",&L,&W,&H))
    {
        read();
        init();
        work();
    }
    return ;
}