PAT A1135 Is It A Red-Black Tree (30 分)
原题链接
- 大致题意
- 思路
- AC代码
1.大致题意
本题只是让你判断是不是红黑树,没有进行其他操纵,算是比较简单的了~
直接用dfs遍历,从红黑树的性质出发就欧克~
给出前序遍历序列,判断一棵树是否是红黑树,树上所有权值都是正整数,前序遍历序列中,负数表示该点是红色的,整数表示该点是黑色的
2.思路
对于任何有效的红黑树有如下性质:
(1)节点是红色或者黑色
(2)根是黑色
(3)所有叶子结点都是黑色
(4)每个红色结点必须有两个黑色的子节点
(从每个叶子到根的所有路径上不能有两个连续的红色结点)
(5)从任一结点到其每个叶子的所有简单路径都包含相同数目的
黑色结点
直接按照要求判断即可~~~
建树过程其实是建立一棵二叉搜索树(左子树小于根节点,右子树大于根节点)
3.AC代码
#include<bits/stdc++.h>
using namespace std;
struct node
{
int val;//权值
struct node *left,*right;
//左右孩子
};
vector<int>arr;
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);
}
/**
getNum()函数和judeg2()函数的目的:
判断条件“从任意结点到叶?结点的路径中,??结点的个数是否相同”
从根节点开始,递归遍历,检查每个结点的左?树的?度和右?树的?度
(这?的?度指??结点的个数),?较左右孩??度是否相等,递归返回结果
红?树不是?种
AVL树,红?树相对于AVL树来说,牺牲了部分平衡性以换取插?/删除操作
时少量的旋转操作,整体来说性能要优于AVL树
**/
int getNum(node *root)
{
if(root==NULL)return 0;
int l=getNum(root->left);
int r=getNum(root->right);
return root->val>0?max(l,r)+1:max(l,r);
}
bool judge2(node*root)
{
if(root==NULL)return true;
int l=getNum(root->left);
int r=getNum(root->right);
if(l!=r) return false;
return judge2(root->left)&&judge2(root->right);
}
int main()
{
#ifdef ONLINE_JUDGE
#else
{
freopen("12.txt","r",stdin);
}
#endif
int k,n;
scanf("%d",&k);
for(int i=0;i<k;i++)
{
scanf("%d",&n);
arr.resize(n);
node *root=NULL;
//每个查询都建立一棵红黑树来判断一下下
for(int j=0;j<n;j++)
{
scanf("%d",&arr[j]);
root=build(root,arr[j]);
}
/**
(1)arr[0]<0: 根节点是红色
(2)judge1(root)==false: 不满足条件“如果当前结点是红?
,它的孩?节点为??”
(3)judge2(root)==false: 不满足条件“从任意结点到叶?结
点的路径中,??结点的个数相同”
**/
if(arr[0]<0||judge1(root)==false||judge2(root)==false)
printf("No\n");
else printf("Yes\n");
}
return 0;
}
喵~