题目大意:
判断一个完全二叉树是不是堆,是最大堆还是最小堆,并且从右到左输出它的每一条路径。
解题思路:
DFS遍历+数组记录路径即可。
代码如下:
#include<iostream>
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
int n,data[1100];
vector<int> v;
bool flagma,flagmi,flagnot;//堆类型
void dfs(int p)
{
if(p*2>n)//边界,叶节点
{
v.push_back(data[p]);//输出路径
printf("%d",v[0]);
for(int i=1;i<v.size();i++)printf(" %d",v[i]);
cout<<endl;
return;
}
v.push_back(data[p]);
if((p*2+1)<=n)//先走右边
{
if(data[p]<=data[p*2+1])//判断堆类型
{
if(!flagma)flagmi=true;
else flagnot=true;//出现冲突
}
else
{
if(!flagmi)flagma=true;
else flagnot=true;
}
dfs(p*2+1);
v.pop_back();
}
if(p*2<=n)
{
if(data[p]<=data[p*2])
{
if(!flagma)flagmi=true;
else flagnot=true;
}
else
{
if(!flagmi)flagma=true;
else flagnot=true;
}
dfs(p*2);
v.pop_back();
}
return ;
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)scanf("%d",&data[i]);
dfs(1);
if(!flagnot)
{
if(flagma)printf("Max Heap\n");
else printf("Min Heap\n");
}
else printf("Not Heap\n");
}