码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2017-04-29 23:24:43      阅读:259      评论:0      收藏:0      [点我收藏+]

标签:memset   sum   越界   结束   ice   size   push   class   iostream   

题意:给定三维数组,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

 

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

标签:memset   sum   越界   结束   ice   size   push   class   iostream   

原文地址:http://www.cnblogs.com/chenxiwenruo/p/6786453.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!