笛卡尔树是一种特殊的二叉树,其结点包含两个关键字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
思路
k1服从搜索树,k2服从小顶堆
注意:
root结点的左孩子的右孩子的k1要小于root结点的k1,
右孩子的左孩子的k1要大于root结点的k1,
所以递归调用过程k1的取值范围,受上一代和上上代结点影响
代码
#include <bits/stdc++.h>
using namespace std;
struct stu{
int k1,k2;
int left=-1;
int right=-1;
}xx[1001];
int flag=1;
void fun(int root,int mink1,int maxk1){
if(xx[root].k1>=maxk1||xx[root].k1<=mink1) flag=0;
if(xx[root].left!=-1){
if(xx[root].k2>=xx[xx[root].left].k2) flag=0;
else fun(xx[root].left,mink1,xx[root].k1);
}
if(xx[root].right!=-1){
if(xx[root].k2>=xx[xx[root].right].k2) flag=0;
else fun(xx[root].right,xx[root].k1,maxk1);
}
}
int main(){
int n;
cin>>n;
int sum=n*(n-1)/2;
for(int i=0;i<n;i++){
cin>>xx[i].k1>>xx[i].k2>>xx[i].left>>xx[i].right;
if(xx[i].left!=-1) sum-=xx[i].left;
if(xx[i].right!=-1) sum-=xx[i].right;
}
fun(sum,-999999,999999);
if(flag) cout<<"YES"<<endl;
else cout<<"NO"<<endl;
return 0;
}