PTA 7-45 笛卡尔树 （25 分）

7-45 笛卡尔树 （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

AC

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
struct tree
{
	int k1,k2,l,r,f;
}tree[1010];
int treemid[1010];
int n,k1,k2,l,r;
int yes1=1,yes2=1,num=0;
void mid(int root)
{
	if(tree[root].l!=-1)
		mid(tree[root].l);
	treemid[num++]=tree[root].k1;
	if(tree[root].r!=-1)
		mid(tree[root].r);
}
void heap(int root)
{
	if(!yes2)
		return;
	if(tree[root].l!=-1)
	{
		if(tree[tree[root].l].k2<tree[root].k2)
		{
			yes2=0;
			return;
		}
	}
	if(tree[root].r!=-1)
	{
		if(tree[tree[root].r].k2<tree[root].k2)
		{
			yes2=0;
			return;
		}
	}
	if(tree[root].l!=-1)
		heap(tree[root].l);
	if(tree[root].r!=-1)
		heap(tree[root].r);
}
int main()
{
	int root;
	cin>>n;
	for(int i=0;i<n;i++)
	{
		tree[i].f=-1;
	}
	for(int i=0;i<n;i++)
	{
		cin>>k1>>k2>>l>>r;
		tree[i].k1=k1;
		tree[i].k2=k2;
		tree[i].l=l;
		tree[i].r=r;
		if(l!=-1)
			tree[l].f=i;
		if(r!=-1)
			tree[r].f=i;
	}
	for(int i=0;i<n;i++)
	{
		if(tree[i].f==-1)
		{
			root=i;
			break;
		}
	}
	mid(root);
	for(int i=0;i<num-1;i++)
	{
		if(treemid[i]>treemid[i+1])
		{
			yes1=0;
			break;
		}
	}
	heap(root);
	if(yes1&&yes2)
		printf("YES");
	else
		printf("NO");
	return 0;
}