题意:给定二叉树的层次遍历,判定是否是一个堆,若是则判断是一个大顶堆还是小顶堆。
思路:不用重构二叉树,直接根据输入的层次遍历来解,根据题意,该二叉树是一个完全二叉树,完全二叉树有很多性质
1.对于一个完全二叉树而言,若该节点无左孩子,那么此节点一定没有右孩子节点(即为叶子节点),根据性质有孩子的节点有n/2个,n为总的节点的个数;
2.若节点i存在左孩子节点和右孩子节点,2*i为其左孩子节点,2*i+1为右孩子节点;
思路:首先根据第一个节点和第二个节点的大小先预判是大顶堆还是小顶堆,然后遍历从0~n-1个非叶子节点,若当前根节点与左右节点的大小关系发生变化,则表明该树不是堆,否则判断是大顶堆还是小顶堆。
最后这个后序遍历二叉树打印,后序遍历二叉树其根节点0一定是最后访问的,故可以判断是否到树的根节点了
#include<iostream>
#include<vector>
using namespace std;
int n = 0, m = 0;
vector<int> p;
void postOrder(int index) {
if (index >= n)return;
postOrder(index * 2 + 1);
postOrder(index * 2 + 2);
printf("%d%s",p[index],index==0?"\n":" ");
}
int main() {
cin >> m >> n;
p.resize(n+1);//分配n+1个空间
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) scanf("%d",&p[j]);
int flag = p[0] > p[1] ? 1 : -1;//1表示大顶堆,-1表示小顶堆
for (int j = 0; j <= n/2-1; j++) {//只需遍历有孩子节点的节点
int left = j * 2 + 1, right = j * 2 + 2;
if (flag == 1 && (left < n&&p[j] < p[left] || (right < n&&p[j] < p[right])))flag = 0;
if (flag == -1 && (left < n&&p[j] > p[left] || (right < n&&p[j] > p[right])))flag = 0;
}
if (flag == 0)printf("Not Heap\n");
else printf("%s\n",flag==1?"Max Heap":"Min Heap");
postOrder(0);
}
system("pause");
return 0;
}