已知一个二叉树，判断它是否为二叉堆(小根堆)

二叉树的节点数N和N个节点(按层输入)

YES或NO

样例输入1

3

1 4 9

样例输入2

3

6 4 9

样例输出1

YES

样例输出2

NO

对于20%的数据  N≤20

对于50%的数据  N≤1000

对于100%的数据 N≤50000,每个节点≤10000

 1 #include <cstdlib>
 2 #include <cstdio>
 3 #include <iostream>
 4 
 5 using namespace std;
 6 
 7 int n,tree[100001],bb,n1;
 8 
 9 int pd()//是否为小根堆 
10 {
11     for (int i=n1-1; i>=2; i--)
12     {
13         if (tree[i]>tree[i/2] && !bb) return 0;//当n为0时 
14         if (tree[i]<tree[i/2] && bb) return 0;
15     }
16     return 1;
17 }
18 
19 int main()
20 {
21     scanf("%d",&n);
22     n1=n;
23     for (int i=1; i<=n; i++)
24         scanf("%d",&tree[i]);
25     while (tree[n1]==tree[n1/2]) n1--;
26     if (tree[n1]>tree[n1/2]) bb=1;
27     if (pd()) cout<<"YES";
28     else cout<<"NO";
29     return 0;
30 }