PAT A1135 Is It A Red-Black Tree (30 分)

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;
}

---

喵~