pat甲级1135 Is It A Red-Black Tree （是否是红黑树）

题意：判断是否是红黑树

1.根节点是黑色

2.红色结点的孩子不能是红色结点

3.从根节点到任意叶节点经过的黑色结点数量相同（下称黑高）

 思路：

1.利用先序序列和二叉查找树性质建树（红黑树也是二叉查找树）

2.遍历树判断红色结点的孩子是否红色结点

3.遍历树，遍历到叶子结点时，将黑高存在set中，最后判断set中是否只有一个黑高

#include <bits/stdc++.h>
typedef long long ll ;
using namespace std ;
const int maxn = 3e1 + 10 ;
const int inf = 0x3f3f3f3f ;
int a[maxn] ;
set<int > st ;//存放树的黑高 
struct node {
    int val;
    struct node *left, *right;
};
node* build(node *root , int v){//红黑树也是二叉查找树 ,根据二叉查找树性质建树 
	if(root == NULL){
		root = new node() ;
		root->val = v ;
		root->left = root->right = NULL ;
	}else if(abs(v) <= abs(root->val)){
	//左子树比根节点小 
		root->left = build(root->left , v) ;
	}else{
	//右子树比根节点大 
		root->right = build(root->right , v) ;
	}
	return root ;
}
bool judge1(node* root){//判断红节点的孩子是否是黑节点 
	if(root == NULL)return true ; 
	if(root->val < 0){
		if(root->left != NULL && root->left->val < 0)return false ;
		if(root->right != NULL && root->right->val < 0)return false ;
	} 
	return judge1(root->left) && judge1(root->right) ;
} 
void judge2(node* root , int h){//计算黑高 
	if(root == NULL){
		st.insert(h) ;
	} else{
		if(root->val > 0){
			judge2(root->left , h+1) ;
			judge2(root->right , h+1) ;
		}else{
			judge2(root->left , h) ;
			judge2(root->right , h) ;
		}
	}
}
void deal(){
	st.clear() ;
	int n ; cin >> n ;
	node *root = NULL ;
	for(int i = 0 ; i < n ; i++){//先序输入 
		cin >> a[i] ;
		root = build(root , a[i]) ; 
	}
	judge2(root , 0) ;
	if(a[0] < 0 || judge1(root) == false || st.size() != 1)
		cout << "No\n" ;
	else 
		cout << "Yes\n" ;
} 
int main(){
	int t ; cin >> t ;
	while(t--)
	deal() ;
    return 0 ;
}