我的笔记之A1155 Heap Paths (30分)

（一次即AC）

题意

给出一个堆，打印出从堆顶到叶子的每条路径（右侧路径优先），并最终判断是否为堆，若是还要判断是大根堆还是小根堆

思路

完全二叉树储存堆，2n为左儿子，2n+1为右儿子，dfs遍历这个堆即可，先dfs (2n+1)，一边遍历一边比较当前节点和上一节点的大小，以及用一个数组存储路径，遍历到叶子时打印该路径

AC代码

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

int tree[1005];
int n;
int minmax;//0:min 1:max
bool flag=true;
vector<int> path;


void dfs(int index){
    path.push_back(tree[index]);
    if(2*index>n&&2*index+1>n){
        printf("%d",tree[1]);
        for(int i=1;i<path.size();i++)
            printf(" %d",path[i]);
        printf("\n");
        path.pop_back();
        return;
    }
    if(2*index+1<=n){
        if(minmax){
            if(tree[2*index+1]>tree[index])   flag=false;
        }else{
            if(tree[2*index+1]<tree[index])   flag=false;
        }
        dfs(2*index+1);
    }
    if(2*index<=n){
        if(minmax){
            if(tree[2*index]>tree[index])   flag=false;
        }else{
            if(tree[2*index]<tree[index])   flag=false;
        }
        dfs(2*index);
    }
    path.pop_back();
}

int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&tree[i]);
    minmax = (tree[1]>tree[2]) ? 1 : 0;
    dfs(1);
    if(flag==false)
        printf("Not Heap");
    else{
        if(minmax==1)
            printf("Max Heap");
        else printf("Min Heap");
    }
    return 0;
}