7-1 笛卡尔树 (25 分)
笛卡尔树是一种特殊的二叉树,其结点包含两个关键字K1和K2。首先笛卡尔树是关于K1的二叉搜索树,即结点左子树的所有K1值都比该结点的K1值小,右子树则大。其次所有结点的K2关键字满足优先队列(不妨设为最小堆)的顺序要求,即该结点的K2值比其子树中所有结点的K2值小。给定一棵二叉树,请判断该树是否笛卡尔树。
输入格式:
输入首先给出正整数N(≤1000),为树中结点的个数。随后N行,每行给出一个结点的信息,包括:结点的K1值、K2值、左孩子结点编号、右孩子结点编号。设结点从0~(N-1)顺序编号。若某结点不存在孩子结点,则该位置给出−1。
输出格式:
输出YES如果该树是一棵笛卡尔树;否则输出NO。
输入样例1:
6
8 27 5 1
9 40 -1 -1
10 20 0 3
12 21 -1 4
15 22 -1 -1
5 35 -1 -1
输出样例1:
YES
输入样例2:
6
8 27 5 1
9 40 -1 -1
10 20 0 3
12 11 -1 4
15 22 -1 -1
50 35 -1 -1
输出样例2:
NO
这题我没有建树用的是结构体数组,重点是找到根节点root在哪,然后中序遍历判断k1是否递增,还要判断k2是否为最小堆
#include<bits/stdc++.h>
using namespace std;
struct Node{
int k1;
int k2;
int left_c;
int right_c;
}a[1200];
int root,flag=1;
int b[1200];
int zhongxu[1200],cnt=0;
void midorder(int root){
if(root!=-1){
midorder(a[root].left_c);
zhongxu[cnt]=a[root].k1;
cnt++;
midorder(a[root].right_c);
}
}
void judgeheap(int root){//判断是不是最小堆
int left,right;
if(a[root].left_c!=-1){
left=a[root].left_c;
if(a[left].k2<a[root].k2){
//如果左孩子的K2比父亲的K2小则不符合最小堆
flag=0;
return ;
}
judgeheap(left);
}
if(a[root].right_c!=-1){
right=a[root].right_c;
if(a[right].k2<a[root].k2){//如果右孩子的K2比父亲的K2小则不符合最小堆
flag=0;
return ;
}
judgeheap(right);
}
}
int main()
{
int n;
cin>>n;
int i,K1,K2,Left,Right;
memset(b,0,sizeof(b));
for(i=0;i<n;i++)
{
scanf("%d %d %d %d",&K1,&K2,&Left,&Right);
a[i].k1=K1;
a[i].k2=K2;
a[i].left_c=Left;
a[i].right_c=Right;
if(Left!=-1)
b[Left]=1;
if(Right!=-1)
b[Right]=1;
}
for(i=0;i<n;i++){
if(b[i]==0){
root=i;
break;
}
}
midorder(root);//中序遍历
for(i=1;i<cnt;i++){
if(zhongxu[i]<=zhongxu[i-1]){
flag=0;
break;
}
}
judgeheap(root);
if(flag)
printf("YES\n");
else
printf("NO\n");
return 0;
}