题目
原题链接:
题目描述
解析
判断一串数,是否构成大顶堆、小顶堆或不是堆
并按从右子树到左子树的深度优先遍历顺序输出所有遍历的情况
代码
#include<stdio.h>
#include<vector>
using namespace std;
vector<int> heap;
int n;
int a[1001];
int isMin = 1, isMax = 1;
void dfs(int index){
if(index*2>n && index*2+1>n){
//判断是否到达叶子节点
if(index<=n){
for(int i=0;i<heap.size();i++){
printf("%d%s",heap[i],i==heap.size()-1?"\n":" ");
}
}
}else{
//右子树入栈进行深度遍历
heap.push_back(a[index*2+1]);
dfs(index*2+1);
heap.pop_back();
//左子树入栈进行深度遍历
heap.push_back(a[index*2]);
dfs(index*2);
heap.pop_back();
}
}
int main(){
// int n;
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
heap.push_back(a[1]);
//有一个违反大/小顶堆的情况,则将对应的flag修改为0
for(int i=2;i<=n;i++){
if(a[i/2]>a[i]) isMin=0;
if(a[i/2]<a[i]) isMax=0;
}
dfs(1);
if(isMin == 1) printf("Min Heap\n");
else if(isMax == 1) printf("Max Heap\n");
else printf("Not Heap\n");
return 0;
}