本题仅仅用到堆的性质(大顶堆和小顶堆)并不涉及对堆的代码考察,实际上考察的还是完全二叉树的性质,以及通过层序遍历得到后序遍历
之前也认真写过一次,但是没有写出来;这次又试了试,过了
对于遍历,还是先记录过程中的数据;最后在main()函数里输出结果
#include <cstdio>
#include <vector>
using namespace std;
const int maxn=1e3+10;
int n, layer[maxn];
vector<int> v;
bool judgemax(){
for(int i=1; i<=n; i++){
if(2*i<=n){
if(layer[2*i]>layer[i])
return false;
}
if(2*i+1<=n){
if(layer[2*i+1]>layer[i])
return false;
}
}
return true;
}
bool judgemin(){
for(int i=1; i<=n; i++){
if(2*i<=n){
if(layer[2*i]<layer[i])
return false;
}
if(2*i+1<=n){
if(layer[2*i+1]<layer[i])
return false;
}
}
return true;
}
void dfs(int root){
if(root>n)
return;
dfs(2*root);
dfs(2*root+1);
v.push_back(layer[root]);
}
int main()
{
int m;
scanf("%d%d", &m, &n);
while(m--){
v.clear();
for(int i=1; i<=n; i++)
scanf("%d", &layer[i]);
if(judgemax()){
printf("Max Heap\n");
dfs(1);
for(int i=0; i<v.size(); i++){
printf("%d", v[i]);
printf(i+1!=v.size() ? " ": "\n");
}
} else if(judgemin()){
printf("Min Heap\n");
dfs(1);
for(int i=0; i<v.size(); i++){
printf("%d", v[i]);
printf(i+1!=v.size() ? " ": "\n");
}
} else {
printf("Not Heap\n");
dfs(1);
for(int i=0; i<v.size(); i++){
printf("%d", v[i]);
printf(i+1!=v.size() ? " ": "\n");
}
}
}
return 0;
}