PAT甲级 1155 Heap Paths DFS+树的层次遍历

最新推荐文章于 2023-03-25 22:57:32 发布

化身孤岛的鲸o

最新推荐文章于 2023-03-25 22:57:32 发布

阅读量177

点赞数

分类专栏： PAT DFS 树文章标签： PAT甲级

本文链接：https://blog.csdn.net/weixin_44123362/article/details/100611469

版权

PAT 同时被 3 个专栏收录

141 篇文章 0 订阅

订阅专栏

树

39 篇文章 0 订阅

订阅专栏

DFS

22 篇文章 0 订阅

订阅专栏

在这里插入图片描述

Solution：

//DFS+树的层次遍历

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;

struct tree{
    int key;
    int left;
    int right;
    tree(){
        left=right=0;
    }
};

vector<tree> t;
int n;
bool flag1=true,flag2=true;
vector<int> path;

void judge_max(int root){
    if(t[root].left==0&&t[root].right==0){
        return;
    }
    int key=t[root].key;
    if(t[root].left==0){
        if(t[t[root].right].key>key){
            flag1=false;
            return;
        }
    }else if(t[root].right==0){
        if(t[t[root].left].key>key){
            flag1=false;
            return;
        }
    }else{
        if(t[t[root].left].key>key||t[t[root].right].key>key){
            flag1=false;
            return;
        }
    }
    if(t[root].left!=0){
        judge_max(t[root].left);
    }
    if(t[root].right!=0){
        judge_max(t[root].right);
    }
}

void judge_min(int root){
    if(t[root].left==0&&t[root].right==0){
        return;
    }
    int key=t[root].key;
    if(t[root].left==0){
        if(t[t[root].right].key<key){
            flag2=false;
            return;
        }
    }else if(t[root].right==0){
        if(t[t[root].left].key<key){
            flag2=false;
            return;
        }
    }else{
        if(t[t[root].left].key<key||t[t[root].right].key<key){
            flag2=false;
            return;
        }
    }

    if(t[root].left!=0){
        judge_min(t[root].left);
    }
    if(t[root].right!=0){
        judge_min(t[root].right);
    }

}

void dfs(int root){
    path.push_back(t[root].key);
    if(t[root].left==0&&t[root].right==0){
        for(int i=0;i<path.size();i++){
            cout<<path[i];
            if(i!=path.size()-1){
                cout<<" ";
            }
        }
        cout<<endl;
        path.pop_back();
    }
    if(t[root].right!=0){
        dfs(t[root].right);
    }
    if(t[root].left!=0){
        dfs(t[root].left);
        path.pop_back();
    }
}

int main(){
    cin>>n;
    t.resize(n);

    for(int i=0;i<n;i++){
        cin>>t[i].key;
    }

    for(int i=0;i<n;i++){
        if(i*2+1<n){
            t[i].left=i*2+1;
        }
        if(i*2+2<n){
            t[i].right=i*2+2;
        }
    }

    judge_max(0);
    judge_min(0);

    dfs(0);

    if(flag1){
        cout<<"Max Heap";
    }else if(flag2){
        cout<<"Min Heap";
    }else{
        cout<<"Not Heap";
    }

    return 0;
}