PAT 1135. Is It A Red-Black Tree (30) 二叉搜索树建立 + 红黑树判断

今天PAT考完试，只做出了3道题，70分。

问题在于读题。

前两题还挺顺利，很快凭借直接做完了，花了50分钟。

第三题卡在题意误解上。  判断的是“各边是否都与 所给集合点 相连” ，而非“各点与各点”相连。 当发现真正题意后，很快就做对了。

第四题最惨，1个小时的时间，有半小时在理解题意上，很短的题目，竟然没注意到“搜索树构建”，只看到“前序序列”。。。

还有，最后其实已经写成功了，就差了一句代码，就是重置根节点——因为这一次有多个输入样例！故搜索树必须重置

只能说命运可能就差了那么一两句代码？

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<string.h>
#include<cmath>
#include<string>
using namespace std;


//跪在题意理解（搜索树构建）和代码错误
/*************************
题意：
构建二叉搜索树，插入的节点有红黑之分
判断该树是否为红黑树。
*************************/

/************************
求解要点：
1.构建搜索树简单，留意data的比较应该用abs(data)比较
2.判断关键
  ①条件1：根节点为黑，即 root->data <0
  ②条件2：先遍历一次，若为红色节点，判断其左右孩子是否为红，若为红，输出No。
  ③条件3：各点到叶结点的黑色节点数相同。
  用递归做，返回左子树的l1，返回右子树的l2，若l1!=l2，说明不平衡，输出No
  若l1等于l2，则把l1 + （当前是否为黑色点） 的值返回到上一层，即return
  当为空节点即黑色叶子节点时，返回1。
3.易错点：
  因为一次有多个输入样例，故必须“重置树”和“重置标志符号flag”！

************************/

/***********************
笔记：
1.关于树的新建问题，为了保险期间
新建参数中命为 Node* &T，因为刚开始root是没有指向的。
*********************/
#define RH -1
#define LH 1
#define EH 0
#define TRUE 1
#define FALSE 0

struct Node{
  int data;
  Node *lchild,*rchild;
};

struct Node *root;//必须为全局


void insert(Node* &T,int data)
{
  if(T==NULL)
  {
    T = (Node *)malloc(sizeof(Node));
    T->data=data;
    T->lchild=T->rchild=NULL;
    return ;
  }

  if(abs(data)<abs(T->data))
  {
    insert(T->lchild,data);
  }
  else insert(T->rchild,data);
}



bool flag;
void juge1(Node *T)
{
  if(flag==false)
    return ;

  if(T->lchild!=NULL)
  {
    if( T->data<0 && T->lchild->data<0)
    {
      flag=false;
      return ;
    }
    else juge1(T->lchild);
  }

  if(T->rchild!=NULL)
  {
    if( T->data<0 && T->rchild->data<0)
    {
      flag=false;
      return ;
    }
    else juge1(T->rchild);
  }
}

int juge2(Node *T)
{
  int l1,l2;
  if(flag==false)
    return -1;

  if(T==NULL)
    return 1;
  
  l1=juge2(T->lchild);
  l2=juge2(T->rchild);
  
  if(l1!=l2)
  {
    flag=false;
    return -1;
  }
  else
  {
    l1=l1 + (T->data>0);
  }

  return l1;
}


int main()
{
  int k,n,data;
  scanf("%d",&k);
  while(k--)
  {
    flag=true;
    root=NULL;
    scanf("%d",&n);
    while(n--)
    {
      scanf("%d",&data);
      insert(root,data);
    }
    if(root->data<0)
      printf("No\n");
    else
    {
      juge1(root);
      if(flag==false)
      {
        cout<<"No"<<endl;
      }
      else
      {
        juge2(root);
        if(flag==false)
          cout<<"No"<<endl;
        else cout<<"Yes"<<endl;
      }
    }
  }
}