题目:https://pintia.cn/problem-sets/15/problems/858
笛卡尔树是一种特殊的二叉树,其结点包含两个关键字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>
#define MAX 1001
typedef struct Tree{
int left,right;
int k1,k2;
}Tree;
int flag[MAX]; //判断根结点,值为1表示为父节点
int judge=1;
Tree t[MAX];
void JudgeHeap(int root){
if(t[root].left!=-1){
if(t[root].k2>=t[t[root].left].k2){
judge=0;
return;
}
JudgeHeap(t[root].left);
}
if(t[root].right!=-1){
if(t[root].k2>=t[t[root].right].k2){
judge=0;
return;
}
JudgeHeap(t[root].right);
}
}
int tree[MAX];
int cnt=0;
void MiddleCreate(int root){ //中序遍历创建树
if(root!=-1){
int tmp=t[root].left;
MiddleCreate(tmp);
tree[cnt++]=t[root].k1;
tmp=t[root].right;
MiddleCreate(tmp);
}
}
int main(){
int n;
scanf("%d",&n);
memset(flag,0,sizeof(flag));
for(int i=0;i<n;i++){
int k1,k2,l,r;
scanf("%d%d%d%d",&k1,&k2,&l,&r);
t[i].k1=k1; t[i].k2=k2;
t[i].left=l; t[i].right=r;
if(l!=-1) flag[l]=1;
if(r!=-1) flag[r]=1;
}
int root;
for(int i=0;i<n;i++){
if(!flag[i]){
root=i;
break;
}
}
JudgeHeap(root);
MiddleCreate(root);
for(int i=0;i<cnt-1;i++){
if(tree[i]>=tree[i+1]){
judge=0;
break;
}
}
if(judge) printf("YES\n");
else printf("NO\n");
}