1135 Is It A Red-Black Tree (30 分)

题目链接
自己写了一个DFS判断的代码，提交上去显示多种错误，我是真的菜。
看了别人的代码，发现自己写递归搜索的时候，总是出现各种问题，很难受。
这个版本是柳神的代码，略做修改，方便以后自己刷第二遍的时候查看。

#include<bits/stdc++.h>
using namespace std;
struct node{
	int data;
	node* left;
	node* right;
};
void build(node* &root, int v){
	if(root==NULL){
		root= new node;
		root->data = v;
		root->left=root->right=NULL;
	}else if(abs(v) < abs(root->data)){
		build(root->left,v);
	}else build(root->right,v); 
}

bool judge1(node *root) {
    if (root == NULL) return true;
    if (root->data < 0) {
        if (root->left != NULL && root->left->data < 0) return false;
        if (root->right != NULL && root->right->data < 0) return false;
    }
    return judge1(root->left) && judge1(root->right);
}
int getNum(node *root) {
    if (root == NULL) return 0;
    int l = getNum(root->left);
    int r = getNum(root->right);
    return root->data > 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(){
	int t,n,num;
	cin>>t;
	while(t--){
		cin>>n;
		node * root = NULL;
		for(int i=0;i<n;i++){
			cin>>num;
			build(root, num);
		}
		if(root->data < 0 || judge1(root)==false ||judge2(root)==false){
			cout<<"No"<<endl;
		}else cout<<"Yes"<<endl;
	}
	return 0;
}