结构与算法 7-31 笛卡尔树 (25 分)（50行精简）

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<string>
#include<algorithm>
using namespace std;
struct Node {
	int K1, K2;
	int left, right;
	bool isRoot = true;
};
vector<Node> tree;
template<typename P>
int getValue(const Node& root, P p) {
	int value = root.K1;
	if (root.left != -1)
		value = max(value, getValue(tree[root.left], p), p);
	if (root.right != -1)
		value = max(value, getValue(tree[root.right], p), p);
	return value;
}
bool isBST(const Node& root){
	if (root.left != -1 && getValue(tree[root.left], less<int>()) > root.K1)
		return false;
	if (root.right != -1 && getValue(tree[root.right], greater<int>()) < root.K1)
		return false;
	return (root.left == -1 || isBST(tree[root.left])) && (root.right == -1 || isBST(tree[root.right]));
}
bool juege(vector<Node> tree) {
	for (Node& node : tree) {
		if (node.left != -1) {		//堆判断
			tree[node.left].isRoot = false;
			if (tree[node.left].K2 <= node.K2)
				return false;
		}
		if (node.right != -1) {
			tree[node.right].isRoot = false;
			if ( tree[node.right].K2 <= node.K2)
				return false;
		}
	}
	int root = 0;
	while (!tree[root].isRoot)	root++;	//寻找根节点
	return isBST(tree[root]);
}

int main() {
	int n;
	cin >> n;
	tree.resize(n);
	for (int i = 0; i < n; i++) 
		cin >> tree[i].K1 >> tree[i].K2 >> tree[i].left >> tree[i].right;
	cout << (juege(tree) ? "YES" : "NO") << endl;
	return 0;
}