这是目前PAT的最后一题了,一开始做的时候,想到了dfs,但是由于只简单学了数据结构,第一次碰到这种类型的题目最开始的想法就是建立线索二叉树,后面又从二叉树性质入手(左右孩子和父节点的关系),利用栈的LIFO做,但是,没能成功实现,最后看了柳神的代码,豁然开朗,STL太好用了!
#include<cstdio>
#include<iostream>
#include<vector>
using namespace std;
int N;
vector<int> treekey;
int a[10010];
void dfs(int index){
if(index*2 > N){
if(index <= N)
//特判最后一个叶子结点是左叶子结点的情况(不理解就画递归流程图)
for(int i = 0;i < treekey.size();i++)
printf("%d%s",treekey[i],i < treekey.size()-1?" ":"\n");
}
else{
treekey.push_back(a[index*2+1]);
dfs(index*2+1);
treekey.pop_back();
treekey.push_back(a[index*2]);
dfs(index*2);
treekey.pop_back();
}
}
int main(){
int isMin = 1,isMax = 1;
cin >> N;
for(int i = 1;i <= N;i++)
cin >> a[i];
treekey.push_back(a[1]);
dfs(1);
for(int i = 2;i <=N;i++){
if(a[i/2] > a[i])
isMin = 0;
if(a[i/2] < a[i])
isMax = 0;
}
if(isMin == 1)
printf("Min Heap");
else
printf("%s",isMax == 1?"Max Heap":"Not Heap");
return 0;
}
上述递归实现如果有点绕的话,建议自己画一个简单的二叉树走一下递归过程!