1135 Is It A Red-Black Tree （红黑树）

#include<bits/stdc++.h>
using namespace std;
const int N=55;
int cnt;
int root;
struct node{
	int val;
	int l,r;
}a[N];
void add(int u,int x){
	if(abs(x)<=abs(a[u].val)){
		if(a[u].l==0){
			a[u].l=cnt;
			a[cnt].val=x;
			cnt++;
		}
		else add(a[u].l,x);
	}
	else{
		if(a[u].r==0){
			a[u].r=cnt;
			a[cnt].val=x;
			cnt++;
		}
		else add(a[u].r,x);
	}
}
bool check1(int u){
	if(u==0) return true;
	if(a[u].val<0){
		if(a[a[u].l].val<0||a[a[u].r].val<0) return false;
	}
	return check1(a[u].l)&&check1(a[u].r);	
}
int get(int u){
	if(u==0) return 0;
	int l=0,r=0;
	l=get(a[u].l);
	r=get(a[u].r);
	return a[u].val>0?max(l,r)+1:max(l,r);  
}
bool check2(int u){
	if(u==0) return true;
	int l=get(a[u].l),r=get(a[u].r);
	if(l!=r) return false;
	return check2(a[u].l)&&check2(a[u].r);
}
int main(){
	int T;scanf("%d",&T);
	while(T--){
		root=1;
		cnt=2;
		int n;scanf("%d",&n);
		for(int i=0;i<n;i++){
			int x;scanf("%d",&x);
			if(i==0) a[root].val=x;
			else add(root,x);
		}
		if(a[root].val<0||!check1(root)||!check2(root)){
			puts("No");
		}
		else puts("Yes");
		for(int i=1;i<=n;i++){
			a[i].val=0;a[i].l=0;a[i].r=0;
		}
	}
	return 0;
}