PAT甲题题解-1091. Acute Stroke (30)-BFS

Posted 2020-09-15 辰曦~文若

题意：给定三维数组，0表示正常，1表示有肿瘤块，肿瘤块的区域>=t才算是肿瘤，求所有肿瘤块的体积和

这道题一开始就想到了dfs或者bfs，但当时看数据量挺大的，以为会导致栈溢出，所以并没有立刻写，想有没有别的办法。
然而结果是，实在想不出别的办法了，所以还是尝试写写dfs、bfs。

一开始先用了dfs，最后两个样例段错误，估计是栈溢出了。
之所以dfs栈溢出，因为dfs的时候每个状态都会存储在堆栈里，就好比dfs的第一个状态，一直保存到最后整个dfs结束。
而bfs是存储在队列中，每次队列都会有状态取出来，所以栈存储就会少很多。

对于每次bfs，计算肿瘤块体积，如果>=t，才算入总和sum中。
用vis[i][j][k]标记结点是否被访问过，被访问过的就不会再次访问，即不会再加入到队列中去。

PS：这里slice和vis数组开成了62*1288*130，即每个维度都比原来多2，是为了处理数组越界

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
using namespace std;
int m,n,l,t;
bool slice[62][1288][130]; //即输入
bool vis[62][1288][130]; //标记对应的点是否访问过
int cnt=0;
struct Node{
    int i,j,k;
};
int BFS(int a,int b,int c){
    if(vis[a][b][c] || slice[a][b][c]==false)
        return 0;
    Node t;
    int i,j,k;
    int res=0;
    queue<Node>q;
    t.i=a;
    t.j=b;
    t.k=c;
    q.push(t);
    vis[a][b][c]=true;
    while(!q.empty()){
        t=q.front();
        q.pop();
        i=t.i;
        j=t.j;
        k=t.k;
        if(i==0 || i>l || j==0 || j>m || k==0 || k>n)
            continue;
        res++;
        //六个方向
        if(slice[i][j-1][k] && !vis[i][j-1][k]){
            t.i=i;
            t.j=j-1;
            t.k=k;
            q.push(t);
            vis[i][j-1][k]=true;
        }
        if(slice[i][j+1][k] && !vis[i][j+1][k]){
            t.i=i;
            t.j=j+1;
            t.k=k;
            q.push(t);
            vis[i][j+1][k]=true;
        }
        if(slice[i][j][k-1] && !vis[i][j][k-1]){
            t.i=i;
            t.j=j;
            t.k=k-1;
            q.push(t);
            vis[i][j][k-1]=true;
        }
        if(slice[i][j][k+1] && !vis[i][j][k+1]){
            t.i=i;
            t.j=j;
            t.k=k+1;
            q.push(t);
            vis[i][j][k+1]=true;
        }
        if(slice[i+1][j][k] && !vis[i+1][j][k]){
            t.i=i+1;
            t.j=j;
            t.k=k;
            q.push(t);
            vis[i+1][j][k]=true;
        }
        if(slice[i-1][j][k] && !vis[i-1][j][k]){
            t.i=i-1;
            t.j=j;
            t.k=k;
            q.push(t);
            vis[i-1][j][k]=true;
        }
    }
    return res;
}
int main()
{
    memset(vis,false,sizeof(vis));
    memset(slice,false,sizeof(slice));
    int a;
    scanf("%d %d %d %d",&m,&n,&l,&t);
    for(int i=1;i<=l;i++){
        for(int j=1;j<=m;j++){
            for(int k=1;k<=n;k++){
                scanf("%d",&a);
                if(a==1)
                    slice[i][j][k]=true;
                else
                    slice[i][j][k]=false;
            }
        }
    }

    int sum=0;
    for(int i=1;i<=l;i++){
        for(int j=1;j<=m;j++){
            for(int k=1;k<=n;k++){
                cnt=BFS(i,j,k);
                if(cnt>=t)
                    sum+=cnt;
            }
        }
    }
    printf("%d\n",sum);
    return 0;
}

View Code