7-31 笛卡尔树 (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

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;


struct Node {
	int k1, k2, lc, rc;
};
vector<Node>tree;
vector<int>v, v1;
int N, k1, k2, lc, rc, root = -1, vis[10001];

void dfs(int x) {
	if (x == -1)
		return;
	dfs(tree[x].lc);
	v.push_back(tree[x].k1), v1.push_back(tree[x].k1);
	dfs(tree[x].rc);
}

int judgek1() {
	dfs(root);
	sort(v.begin(), v.end());
	for (int i = 0; i < v.size(); ++i)
		if (v[i] != v1[i])
			return 0;
	return 1;
} 

int judgek2() {
	for (auto& node : tree) {
		if (node.lc != -1)
			if (tree[node.lc].k2 <= node.k2)
				return 0;
		if (node.rc != -1)
			if (tree[node.rc].k2 <= node.k2)
				return 0;
	}
	return 1;
}

int main() {
	cin >> N;
	for (int i = 0; i < N; ++i) {
		cin >> k1 >> k2 >> lc >> rc;
		tree.push_back({ k1,k2,lc,rc });
		vis[lc] = (lc == -1 ? 0 : 1);
		vis[rc] = (rc == -1 ? 0 : 1);
	}
	while (vis[++root]);
	cout << (judgek2() && judgek1() ? "YES" : "NO");
	return 0;
}