PAT A1135 Is It A Red-Black Tree (30分)红黑树判断

维基百科：

红黑树（英语：Red–black tree）是一种自平衡二叉查找树，是在计算机科学中用到的一种数据结构，典型的用途是实现关联数组。它在1972年由鲁道夫·贝尔发明，被称为"对称二叉B树"，它现代的名字源于Leo J. Guibas和Robert Sedgewick于1978年写的一篇论文。红黑树的结构复杂，但它的操作有着良好的最坏情况运行时间，并且在实践中高效：它可以在O(log n)时间内完成查找、插入和删除，这里的n是树中元素的数目。

红黑树是每个节点都带有颜色属性的二叉查找树，颜色为红色或黑色。在二叉查找树强制一般要求以外，对于任何有效的红黑树我们增加了如下的额外要求：

1. 节点是红色或黑色
2. 根是黑色。
3. 所有叶子都是黑色（叶子是NIL节点）。
4. 每个红色节点必须有两个黑色的子节点。（从每个叶子到根的所有路径上不能有两个连续的红色节点。）
5. 从任一节点到其每个叶子的所有简单路径都包含相同数目的黑色节点。

以下参考柳神的代码

#include <iostream>
#include <cmath>
using namespace std;

struct node{
	int data;
	node *lchild, *rchild;
};

//根据二叉搜索树的先序序列建立这棵二叉搜索树
void Create(node *&root, int x){
	if(root == NULL){
		root = new node;
		root->data = x;
		root->lchild = NULL;
		root->rchild = NULL;
		return;
	}
	if(abs(x) <= abs(root->data)){
		Create(root->lchild, x);
	}else {
		Create(root->rchild, x);
	}
}

//判断红色结点的孩子是否是黑色结点
bool judge1(node *root){
	if(root == NULL)	return true;
	if(root->data < 0){
		if(root->lchild != NULL && root->lchild->data < 0){
			return false;
		}
		if(root->rchild != NULL && root->rchild->data < 0){
			return false;
		}
	}
	return judge1(root->lchild) && judge1(root->rchild);	//
}

//计算根结点的左右子树中黑色结点的最大个数，类似于AVL树中的高度
int getNum(node* root){
	if (root == NULL)	return 0;
	int left = getNum(root->lchild);
	int right = getNum(root->rchild);
	return root->data > 0 ? max(left, right) + 1 : max(left, right);
}

//函数判断某个结点的左右子树中黑结点的个数（类比AVL树的高度）是否相等
bool judge2(node* root){
	if(root == NULL)	return true;	//结点为空，返回true
	int left = getNum(root->lchild);	//计算左子树中的黑结点个数
	int right = getNum(root->rchild);	//计算右子树中的黑结点个数
	if(left != right)	return false;	//如果左右子树中黑结点个数不相等则返回false
	return judge2(root->lchild) && judge2(root->rchild);	//如果相等，向左右子树递归，返回与的结果
}

int main(){
	int n;
	scanf("%d", &n);
	for(int i = 0; i < n; i++){
		int m;
		scanf("%d", &m);
		node *root = NULL;
		for(int j = 0; j < m; j++){
			int num;
			scanf("%d", &num);
			Create(root, num);
		}
		if(root->data < 0 || judge1(root) == false || judge2(root) == false){		
			printf("No\n");
		}else{
			printf("Yes\n");
		}
	}
	return 0;	
}