PAT A1135 Is It A Red Black Tree

判断一棵树是否是红黑树，按题给条件建树，dfs判断即可~

#include<bits/stdc++.h>
using namespace std;
const int maxn=;
struct node {
    int data;
    node * left=NULL;
    node * right=NULL;
};
void insert (node * &root,int x) {
    if (root==NULL) {
        root=new node;
        root->data=x;
        return;
    }
    if (abs(x)<abs(root->data)) insert (root->left,x);
    else insert (root->right,x);
}
int getnum (node * root) {
    if (root==NULL) return ;
    if (root->data>)
    return max(getnum(root->left),getnum(root->right))+;
    else return max(getnum(root->left),getnum(root->right));
}
int flag=;
void dfs (node * root) {
    if (root==NULL) return;
    if (root->data<&&root->left&&root->left->data<) flag++;
    if (root->data<&&root->right&&root->right->data<) flag++;
    if (getnum(root->left)!=getnum(root->right)) flag++;
    dfs (root->left);
    dfs (root->right);
}
int main () {
    int T;
    scanf ("%d",&T);
    int N,x;
    while (T--) {
        node * root=NULL;
        scanf ("%d",&N);
        for (int i=;i<N;i++) {
            scanf ("%d",&x);
            insert (root,x);
        }
        flag=;
        if (root->data<) flag++;
        dfs (root);
        if (flag==) printf ("Yes\n");
        else printf ("No\n");
    }
    return ;
}