1135 Is It A Red-Black Tree （30 分）

问题描述：给定一颗树的先序，判断是否为红黑树。

解题思路：满足BST，直接插入建树，判断需每个节点都判断，按理来说，只判断根节点到所有叶节点的黑节点个数是否相等即可呀？？？？？？？

AC代码：

#include<iostream>
#include<unordered_map>
#include<algorithm>
#include<cmath>
using namespace std;
typedef struct node{
    int left,right;
    node(){}//需要这个，不然会出错！！！！！
    node(int l,int r):left(l),right(r){}
}node;
int M=0;
unordered_map<int,node>arr;
void BSTinsert(int &v,int x)//建树，v相当于指针，
{
    if(!v){v=x;return;}
    if(abs(v)<abs(x)){//往右子树插
        BSTinsert(arr[v].right,x);
    }else{//左子树插
        BSTinsert(arr[v].left,x);
    }
}
/*void pre(int v)
{
    if(!v)return;
    printf("%d ",v);
    pre(arr[v].left);
    pre(arr[v].right);
}*/
/*void judge(int v,bool &f,int cnt)//测试点3错误，没有检查其他节点，只检测了根节点。
{
    if(!v){//叶节点
        if(!M)M=cnt;
        else if(M!=cnt)f=1;
        return;
    }else if(v>0){//黑节点
        ++cnt;
    }else if(arr[v].right<0||arr[v].left<0){//红节点
        f=1;return;
    }
    judge(arr[v].left,f,cnt);//左子树
    if(!f)judge(arr[v].right,f,cnt);//右子树

}*/
int judge(int v,bool &f)
{
    if(!v||f)return 1;//叶节点
    int l=judge(arr[v].left,f);
    int r=judge(arr[v].right,f);
    if(l!=r)f=1;//左右黑节点个数不一样
    if(v<0&&(arr[v].right<0||arr[v].left<0))f=1;//红节点
    if(v>0)++l;//黑节点
    return l;
}

int main()
{
    freopen("test.txt","r",stdin);
    int K,N,root,a;
    bool flag=0;
    scanf("%d",&K);
    while(K--){
        arr.clear();flag=0;
        scanf("%d %d",&N,&root);
        if(root<0)flag=1;//根为红
        arr[root]=node(0,0);
        while(--N){
            scanf("%d",&a);
            if(flag)continue;
            arr[a]=node(0,0);
            BSTinsert(root,a);
        }
       // pre(root);printf("\n");
        if(!flag)judge(root,flag);
        if(!flag)printf("Yes\n");
        else printf("No\n");
    }
    return 0;
}