PAT (Advanced Level) Practice - 1135 Is It A Red-Black Tree(30 分)

简介: PAT (Advanced Level) Practice - 1135 Is It A Red-Black Tree(30 分)

题目链接:点击打开链接

题目大意:给一棵二叉搜索树的前序遍历,判断它是否为红黑树,是输出Yes,否则输出No。

解题思路:

判断以下几点:

  1. 根结点是否为黑色;
  2. 如果一个结点是红色,它的孩子节点是否都为黑色;
  3. 从任意结点到叶子结点的路径中,黑色结点的个数是否相同。

所以分为以下几步:

  1. 根据先序建立一棵树,用链表表示
  2. 判断根结点(题目所给先序的第一个点即根结点)是否是黑色【v[0] < 0】
  3. 根据建立的树,从根结点开始遍历,如果当前结点是红色,判断它的孩子节点是否为黑色,递归返回结果【jde1函数】
  4. 从根节点开始,递归遍历,检查每个结点的左子树的高度和右子树的高度(这里的高度指黑色结点的个数),比较左右孩子高度是否相等,递归返回结果【jde2函数】

注意:维基百科定义:红黑树(英语:Red–black tree)是一种自平衡二叉查找树。AVL树:在计算机科学中,AVL树是最先发明的自平衡二叉查找树。在AVL树中任何节点的两个子树的高度最大差别为1,所以它也被称为高度平衡树。所以说红黑树不是一种AVL树,红黑树相对于AVL树来说,牺牲了部分平衡性以换取插入/删除操作时少量的旋转操作,整体来说性能要优于AVL树。而我根据先序遍历直接建树后判断了是否AVL平衡,把判断是否平衡的那段代码注释掉就AC了~


AC 代码

#include<bits/stdc++.h>#include<cmath>#define mem(a,b) memset(a,b,sizeof a)#define ssclr(ss) ss.clear(), ss.str("")#define INF 0x3f3f3f3f#define MOD 1000000007usingnamespacestd;
typedeflonglongll;
structnode{
intd;
node*left, *right;
};
vector<int>v;
node*insertRBT(node*rt, intval)
{
if(rt==NULL)
    {
rt=newnode();
rt->d=val;
rt->left=rt->right=NULL;
    }
elseif(abs(rt->d)>=abs(val)) rt->left=insertRBT(rt->left,val);
elsert->right=insertRBT(rt->right,val);
returnrt;
}
intgetNum(node*rt)
{
if(rt==NULL) return0;
intl=getNum(rt->left), r=getNum(rt->right);
returnrt->d>0?max(l,r)+1 : max(l,r);
}
intjde1(node*rt) // 判断结点为red时,孩子是否为black{
if(rt==NULL) return1;
if(rt->d<0)
    {
if(rt->left!=NULL&&rt->left->d<0) return0;
if(rt->right!=NULL&&rt->right->d<0) return0;
    }
returnjde1(rt->left) &&jde1(rt->right);
}
intjde2(node*rt) // 判断黑结点数是否相等{
if(rt==NULL) return1;
intl=getNum(rt->left), r=getNum(rt->right);
if(l!=r) return0;
returnjde2(rt->left) &&jde2(rt->right);
}
intmain()
{
intT; scanf("%d",&T);
while(T--)
    {
intn;
scanf("%d",&n);
node*tree=NULL;
v.resize(n);
for(inti=0;i<n;i++)
        {
scanf("%d",&v[i]);
tree=insertRBT(tree,v[i]);
        }
if(v[0]<0||!jde1(tree) ||!jde2(tree)) puts("No");
elseputs("Yes");
    }
return0;
}
目录
相关文章
PAT (Advanced Level) Practice - 1045 Favorite Color Stripe(30 分)
PAT (Advanced Level) Practice - 1045 Favorite Color Stripe(30 分)
83 0
PAT (Advanced Level) Practice - 1045 Favorite Color Stripe(30 分)
PAT (Advanced Level) Practice - 1057 Stack(30 分)
PAT (Advanced Level) Practice - 1057 Stack(30 分)
96 0
PAT (Advanced Level) Practice - 1057 Stack(30 分)
PAT (Advanced Level) Practice - 1091 Acute Stroke(30 分)
PAT (Advanced Level) Practice - 1091 Acute Stroke(30 分)
85 0
PAT (Advanced Level) Practice - 1053 Path of Equal Weight(30 分)
PAT (Advanced Level) Practice - 1053 Path of Equal Weight(30 分)
105 0
PAT (Advanced Level) Practice - 1127 ZigZagging on a Tree(30 分)
PAT (Advanced Level) Practice - 1127 ZigZagging on a Tree(30 分)
106 0
|
算法
PAT (Advanced Level) Practice - 1033 To Fill or Not to Fill(25 分)
PAT (Advanced Level) Practice - 1033 To Fill or Not to Fill(25 分)
78 0
PAT (Advanced Level) Practice - 1143 Lowest Common Ancestor(30 分)
PAT (Advanced Level) Practice - 1143 Lowest Common Ancestor(30 分)
99 0
PAT (Advanced Level) Practice - 1068 Find More Coins(30 分)
PAT (Advanced Level) Practice - 1068 Find More Coins(30 分)
97 0
|
存储 测试技术
PAT (Advanced Level) Practice - 1131 Subway Map(30 分)
PAT (Advanced Level) Practice - 1131 Subway Map(30 分)
94 0
|
存储
PAT (Advanced Level) Practice - 1126 Eulerian Path(25 分)
PAT (Advanced Level) Practice - 1126 Eulerian Path(25 分)
119 0