PAT 甲级 1155 Heap Paths (30 分)

题目

题目跳转 1155 Heap Paths (30 分)

分析

题目大意是判断给定的层级遍历的二叉树是不是堆，是的话是最大堆还是最小堆。这里看完了算法笔记上对堆的分析，有一个要点，就是可以用数组表示堆 用n*2表示左节点2n+1表示右节点。然后就根据定义写30分就到手了。

code

#include <stdio.h>
#include <iostream>
#include <vector>
using namespace std;
const int M = 1001;
int n, nums[M];
vector<vector<int> > allpath;
/** 得到所有的路径 **/
void visitTree(int index, vector<int> path){
    if(index>n||index<1){
        return;
    }else if(index>n/2&&index<=n){
        // 如果是叶子节点就添加后返回
        path.push_back(nums[index]);
        allpath.push_back(path);
        return;
    }else {
        // 非叶子节点添加进去
        path.push_back(nums[index]);
    }
    int lindex = index*2, rindex = index*2+1;
    visitTree(rindex,path);
    visitTree(lindex,path);
}

int main(){
    cin>>n;
    for (int i = 0; i < n; i++)
    {
        int num;
        cin >> num;
        nums[i+1] = num;
    }
    vector<int> pp;
    // 打印路径
    visitTree(1,pp);
    for (int i = 0; i < allpath.size(); i++)
    {
        vector<int> vec = allpath[i];
        for (int j = 0;j < vec.size();j++)
        {
            if(j) cout<<" ";
            cout<<vec[j];
        }
        cout<<endl;
    }
    int isMin = 1,isMax = 1;
    for (int i = 2; i <= n; i++)
    {
        if(nums[i/2]>nums[i]) isMin = 0;
        if(nums[i/2]<nums[i]) isMax = 0;
    }
    if(isMin){
        cout<<"Min Heap"<<endl;
    }else if(isMax){
        cout<<"Max Heap"<<endl;
    }else{
        cout<<"Not Heap"<<endl;
    }
    
}

结果