dfs每次插入一个当前点,检测到当前节点为叶子节点(u * 2 > n)时输出nodes中的全部元素,每次弹出nodes中的最后一个元素还原现场。判断类型用lt(less than)和gt(greater than)两个变量,每次输出nodes时顺带判断nodes当前元素较上一元素的变换情况。最后如果gt和lt都为true,说明路径的序列有递增也有递减,则不是堆,若只有gt为true,则说明路径是递增的,则为小根堆,否则为大根堆。
#include <bits/stdc++.h>
using namespace std;
const int N = 1010;
int w[N];
int n;
vector<int> nodes;
bool lt, gt;
void dfs(int u)
{
nodes.push_back(w[u]);
if(u * 2 > n)
{
cout << nodes[0];
for(int i = 1; i < nodes.size(); i++)
{
cout << ' ' << nodes[i];
if(nodes[i] > nodes[i - 1]) gt = true;
if(nodes[i] < nodes[i - 1]) lt = true;
}
cout << endl;
}
if(u * 2 + 1 <= n) dfs(u * 2 + 1);
if(u * 2 <= n) dfs(u * 2);
nodes.pop_back();
}
int main()
{
cin >> n;
for(int i = 1; i <= n; i ++) cin >> w[i];
dfs(1);
if(gt && lt) puts("Not Heap");
else if(lt) puts("Max Heap");
else puts("Min Heap");
}