给出m次查询,每次都是n长度的树。每次给出一棵完全二叉树的层次遍历,判断其是否是堆,是最大堆(父节点
>
=
>=
>=子节点)还是最小堆,不是堆输出Not Heap,最大堆输出Max Heap,最小堆Min Heap,然后再输出该树的后序遍历。
先根据前两个元素的值假设该树是最大堆或最小堆,然后直接进行后序遍历,遍历中判断父节点是否符合该假设。
(用时: 25:40.31)
#include <bits/stdc++.h>
using namespace std;
int queSize =0;
int isheap = 0;
void postOrder(int node[],int index,int length,int que[])
{
if(index >= length) {
return;
}
if(2*index+1 < length) {
if(isheap==1) {
if(node[index] >= node[2*index+1]) {
isheap=0;
}
} else if(isheap==2) {
if(node[index] < node[2*index+1]) {
isheap=0;
}
}
postOrder(node,2*index+1,length,que);
}
if(2*index+2 < length) {
if(isheap==1) {
if(node[index] >= node[2*index+2]) {
isheap=0;
}
} else if(isheap==2) {
if(node[index] < node[2*index+2]) {
isheap=0;
}
}
postOrder(node,2*index+2,length,que);
}
que[queSize++] = node[index];
}
int main()
{
int n,m;
scanf("%d%d",&m,&n);
int node[n];
int que[n];
for(int j=0; j<m; j++) {
for(int i=0; i<n; i++) {
scanf("%d",&node[i]);
}
if(node[0] >= node[1]) {
isheap = 2;
} else {
isheap = 1;
}
queSize = 0;
postOrder(node,0,n,que);
switch(isheap) {
case 1:
printf("Min Heap\n");
break;
case 2:
printf("Max Heap\n");
break;
default:
printf("Not Heap\n");
break;
}
for(int i=0; i<n; i++) {
if(i==0) {
printf("%d",que[i]);
} else {
printf(" %d",que[i]);
}
}
printf("\n");
}
return 0;
}