笛卡尔树是一种特殊的二叉树,其结点包含两个关键字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
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
using namespace std;
struct node{
int k1,k2;
int left,right;
}t[1005];
int vis[1005];
int a[1005],b[1005];
int flag1,flag2;
void judge_k1(int root){//对于每一个结点,依次将他的k2与其左右子结点的k2进行比较即可
int left,right;
if(t[root].left!=-1){ //判断完当前结点的左结点后,递归递归左结点
left=t[root].left;
if(t[left].k2<t[root].k2){
flag1=1;
return;
}
judge_k1(left);
}
if(t[root].right!=-1){ //同理右结点;
right=t[root].right;
if(t[right].k2<t[root].k2){
flag1=1;
return;
}
judge_k1(right);
}
}
int cnt;
int inorder(int root){
if(root!=-1){//注意是当前数不为-1;
inorder(t[root].left); //遍历左子树,入数组,遍历右子树;
a[cnt]=t[root].k1;
b[cnt]=t[root].k1;
cnt++;
inorder(t[root].right);
}
}
int main(){
int n;
memset(vis,0,sizeof vis);
scanf("%d",&n);
for(int i=0;i<n;i++){ //将亦将作过别人子结点的,且不为空的结点标记,其不可能成为根节点
scanf("%d%d%d%d",&t[i].k1,&t[i].k2,&t[i].left,&t[i].right);
if(t[i].left!=-1) vis[t[i].left]=1;
if(t[i].right!=-1) vis[t[i].right]=1;
}
int root;
for(int i=0;i<n;i++){
if(vis[i]==0) root=i; //找到根节点即可串联成树
}
judge_k1(root);
inorder(root); //判断k1是否符合二叉搜索树时,只需将其中序遍历的结果和其排序之后的结果比对即可;
sort(a,a+n);
for(int i=0;i<n;i++){
if(a[i]!=b[i]){
flag2=1;
break;
}
}
if(!flag1&&!flag2) printf("YES\n");
else printf("NO\n");
return 0;
}