题目描述:
笛卡尔树是一种特殊的二叉树,其结点包含两个关键字K1和K2。首先笛卡尔树是关于K1的二叉搜索树,即结点左子树的所有K1值都比该结点的K1值小,右子树则大。其次所有结点的K2关键字满足优先队列(不妨设为最小堆)的顺序要求,即该结点的K2值比其子树中所有结点的K2值小。给定一棵二叉树,请判断该树是否笛卡尔树。
输入格式:
输入首先给出正整数N(≤1000),为树中结点的个数。随后N行,每行给出一个结点的信息,包括:结点的K1值、K2值、左孩子结点编号、右孩子结点编号。设结点从0~(N-1)顺序编号。若某结点不存在孩子结点,则该位置给出−1。
输出格式:
输出YES如果该树是一棵笛卡尔树;否则输出NO。
输入样例:
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
输出样例:
YES
判断最小堆的时候,直接模拟判断,判断二叉搜索树的时候不能只是单纯的判断局部的二叉树是二叉搜索树,而是要判断整棵树都是二叉搜索树。判断二叉搜索树时利用它的中序遍历为递增序列。
#include <bits/stdc++.h>
using namespace std;
const int N = 10000;
struct node {
int k1, k2;
int l, r;
} tree[N];
int mid[N], idx;
int in[N];
void midorder(int root) {
if (root != -1) {
midorder(tree[root].l);
mid[idx++] = tree[root].k1;
midorder(tree[root].r);
}
}
int main() {
int n; cin >> n;
for (int i = 0; i < n; i++) {
cin >> tree[i].k1 >> tree[i].k2;
cin >> tree[i].l >> tree[i].r;
if (tree[i].l != -1) in[tree[i].l]++;
if (tree[i].r != -1) in[tree[i].r]++;
}
int flag = 1;
for (int i = 0; i < n; i++) {
int l = tree[i].l, r = tree[i].r;
if (l != -1 && r != -1) {
if (tree[l].k2 <= tree[i].k2) flag = 0;
if (tree[r].k2 <= tree[i].k2) flag = 0;
} else if (l != -1 && r == -1) {
if (tree[l].k2 <= tree[i].k2) flag = 0;
} else if (l == -1 && r != -1) {
if (tree[r].k2 <= tree[i].k2) flag = 0;
}
if (!flag) break;
}
int root = 0;
for (int i = 0; i < n; i++) {
if (in[i] == 0) {
root = i;
break;
}
}
midorder(root);
for (int i = 1; i < n; i++) {
if (mid[i - 1] > mid[i]) flag = 0;
}
if (flag) cout << "YES\n";
else cout << "NO\n";
return 0;
}