1155 Heap Paths (30分)

1155 Heap Paths (30分)

题目要求

从根至叶打印路径
右子树路径优先打出
判断是大顶堆、小顶堆、不是堆

解题思想

a[]从下标1开始存放完全二叉树
v[]从0开始存放路径
使用dfs按要求遍历路径
dfs内的判断条件：
如果当前结点没有孩子结点且位于树内，说明其为路径最后一个结点，此时打印路径
如果当前结点有孩子结点，则将其放入路径，从其右孩子开始继续dfs，结束后将右孩子从路径弹出，再从其左孩子继续dfs，结束后将左孩子从路径弹出
另一种解释：弹出结点目的是不影响后续路径的输出
判断堆使用isMax、isMin
利用完全二叉树数组存放性质
如果孩子结点大于父结点，说明不是大顶堆
如果孩子结点小于父结点，说明不是小顶堆

AC代码

#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
int a[1010], isMax = 1, isMin = 1, n;
vector<int> v;
void dfs(int index){
    if (index*2+1>n && index*2>n){
        if (index <= n){
            for (int i = 0; i < int(v.size()); i++)
                printf("%d%s", v[i], i!=v.size()-1 ? " " : "\n");
        }
    }
    else {
        v.push_back(a[index*2+1]);
        dfs(index*2+1);
        v.pop_back();
        v.push_back(a[index*2]);
        dfs(index*2);
        v.pop_back();
    }
}
int main(){
    scanf("%d", &n);
    for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
    v.push_back(a[1]);
    dfs(1);
    for (int i = 2; i <= n; i++){
        if (a[i/2] > a[i]) isMin = 0;
        if (a[i/2] < a[i]) isMax = 0;
    }
    if (isMax == 1) printf("Max Heap\n");
    else if (isMin == 1) printf("Min Heap\n");
    else printf("Not Heap\n");
    return 0;
}