红黑树的特点
1、根节点是黑色。
2、如果一个节点是红色那么他的两个子节点都是红色。
3、任意从根节点到叶子结点的路径上,所有的路径经过的黑色节点数相同。
4、红黑树是二叉搜索树。
算法
1、根节点是否为黑色。
2、红色节点的两个子节点是不是都是黑色。
3、所有路径经过的黑色节点数是不是相同的。
#include <bits/stdc++.h>
using namespace std;
vector<int> arr;
struct node
{
int key;
struct node *lchild,*rchild;
};
node *build(node *root,int v)
{
if(root==NULL)
{
root = (node*)malloc(sizeof(node));
root->key = v;
root->lchild = root->rchild = NULL;
}else if(abs(v)<=abs(root->key))
{
root->lchild = build(root->lchild,v);
}else
{
root->rchild = build(root->rchild,v);
}
return root;
}
bool judge1(node *root)
{
if(root==NULL) return true;
if(root->key<0)
{
if(root->lchild!=NULL && root->lchild->key<0) return false;
if(root->rchild!=NULL && root->rchild->key<0) return false;
}
return judge1(root->lchild)&&judge1(root->rchild);
}
int getBlackNum(node *root)
{
if(root==NULL) return 0;
int l = getBlackNum(root->lchild);
int r = getBlackNum(root->rchild);
return root->key>0?max(l,r)+1:max(l,r);
}
bool judge2(node *root)
{
if(root==NULL) return true;
int l= getBlackNum(root->lchild);
int r = getBlackNum(root->rchild);
if(l!=r) return false;
return judge2(root->lchild)&&judge2(root->rchild);
}
int main()
{
int k,n;
scanf("%d",&k);
for(int i=0;i<k;i++)
{
node *root=NULL;
scanf("%d",&n);
arr.resize(n);
for(int j=0;j<n;j++)
{
scanf("%d",&arr[j]);
root = build(root,arr[j]);
}
if(arr[0]<0 || judge1(root)==false || judge2(root)==false)
printf("No\n");
else printf("Yes\n");
}
return 0;
}