笛卡尔树是一种特殊的二叉树,其结点包含两个关键字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<stdio.h>
#include<string.h>
#include<stdlib.h>
typedef struct N{
int K1;
int K2;
int Left;
int Right;
} Node;
typedef Node* TreeNode;
void Ergodic(TreeNode tNode[],int pos,int parentK1);
void MidErgodic(TreeNode tNode[],int pos);
int SearchTree[1000]={0};
int Spos=0;
int main(){
int n;
scanf("%d",&n);
TreeNode *tNode=(TreeNode*)malloc(sizeof(TreeNode)*n);
for(int i=0; i<n; i++){
tNode[i]=(TreeNode)malloc(sizeof(Node));
tNode[i]->K1=0;
tNode[i]->K2=0;
tNode[i]->Left=-1;
tNode[i]->Right=-1;
}
for(int i=0; i<n; i++){
scanf("%d %d %d %d",&tNode[i]->K1,&tNode[i]->K2,&tNode[i]->Left,&tNode[i]->Right);
}
int *isV=(int *)malloc(sizeof(int)*n);
for(int i=0; i<n; i++){
isV[i]=0;
}
for(int i=0; i<n; i++){
if(tNode[i]->Left!=-1){
isV[tNode[i]->Left]=1;
}
if(tNode[i]->Right!=-1){
isV[tNode[i]->Right]=1;
}
}
int First=0;
for(int i=0; i<n; i++){
if(!isV[i]){
First=i;
break;
}
}
Ergodic(tNode,First,10000);
MidErgodic(tNode,First);
int flag=1;
for(int i=1; i<Spos; i++){
if(SearchTree[i]<SearchTree[i-1]){
printf("NO\n");
flag=0;
break;
}
}
if(flag)
printf("YES\n");
return 0;
}
void Ergodic(TreeNode tNode[],int pos,int parentK1){
TreeNode root=tNode[pos];
TreeNode Left=NULL;
if(tNode[pos]->Left!=-1){
Left=tNode[tNode[pos]->Left];
}
TreeNode Right=NULL;
if(tNode[pos]->Right!=-1){
Right=tNode[tNode[pos]->Right];
}
if(Left!=NULL && Right!=NULL){
if((root->K1 > Left->K1 && root->K2 < Left->K2)
&& (root->K1 < Right->K1 && root->K2 < Right->K2)
){
Ergodic(tNode,root->Left,root->K1);
Ergodic(tNode,root->Right,root->K1);
}else{
printf("NO\n");
exit(0);
}
}else if(Left!=NULL && Right==NULL){
if(root->K1 > Left->K1 && root->K2 < Left->K2){
Ergodic(tNode,root->Left,root->K1);
}else{
printf("NO\n");
exit(0);
}
}else if(Left==NULL && Right != NULL){
if(root->K1 < Right->K1 && root->K2 < Right->K2){
Ergodic(tNode,root->Right,root->K1);
}else{
printf("NO\n");
exit(0);
}
}
}
void MidErgodic(TreeNode tNode[],int pos){
if(pos!=-1){
TreeNode root=tNode[pos];
MidErgodic(tNode,root->Left);
SearchTree[Spos++]=root->K1;
MidErgodic(tNode,root->Right);
}
}