1135 Is It A Red-Black Tree （30 point(s)）

最新推荐文章于 2024-10-11 09:39:07 发布

徐伯莱

最新推荐文章于 2024-10-11 09:39:07 发布

阅读量343

点赞数

CC 4.0 BY-SA版权

分类专栏： Pat甲级文章标签： Pat甲级

本文链接：https://blog.youkuaiyun.com/m0_37691414/article/details/87526522

Pat甲级专栏收录该内容

152 篇文章

订阅专栏

本文介绍了一种使用C++实现的红黑树性质验证方法，包括构建红黑树、检查红黑树的性质是否满足以及如何计算黑色节点的数量。通过递归算法确保红黑树的性质，如每个节点的子树中红色节点不超过一个黑色节点，所有路径上的黑色节点数量相等。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题解

红黑树性质的判断。

#include<iostream>
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
vector<int> pre;
int n, k;
struct node {
	node *l, *r;
	int v;
};
node* build(node* root, int v) {
	if(root == NULL) {
		root = new node();
		root->l = root->r = NULL;
		root->v = v;
		return root;
	}
	if(abs(v) < abs(root->v)) root->l = build(root->l, v);
	else root->r = build(root->r, v);
	return root;
} 
int getBlackNum(node* root) {
	if(root == NULL) return 0;
	int l = getBlackNum(root->l);
	int r = getBlackNum(root->r);
	return root->v > 0 ? max(l, r) + 1 : max(l, r);
}
bool judge4(node* root) { // 检验第四条性质
	if(root == NULL) return true;
	if(root->v < 0) {
		if(root->l != NULL && root->l->v < 0) return false;
		if(root->r != NULL && root->r->v < 0) return false;
	}
	return judge4(root->l) && judge4(root->r);
}
bool judge5(node* root) { // 检验第五条性质
	if(root == NULL) return true;
	int l = getBlackNum(root->l);
	int r = getBlackNum(root->r);
	if(l != r) return false;
	return judge5(root->l) && judge5(root->r);
}
int main() {
	scanf("%d", &k);
	for(int i = 0; i < k; ++i) {
		scanf("%d", &n); pre.resize(n);
		node* root = NULL;
		for(int j = 0; j < n; ++j) {
			scanf("%d", &pre[j]);
			root = build(root, pre[j]);
		}
		if(pre[0] < 0 || !judge4(root) || !judge5(root)) printf("No\n");
		else printf("Yes\n");
	}
	return 0;
}