PAT 甲级 1135 Is It A Red-Black Tree

题目链接：点击打开链接

题意：给出一棵二叉搜索树，判断是否为红黑树（一种二叉平衡搜索树）。

思路：其实很简单，利用给出的前序遍历的顺序插入节点，可以直接建造出对应的二叉搜索树，再根据红黑树的特征去判断是否为红黑树。

#include <cstdio>
#include <cstring>
#include <cmath>
#include <iostream>
using namespace std;
int data[50],le[50],rig[50];
int root,len;
void Insert(int &ro,int loc){//先建二叉搜索树（按照先序遍历的顺序插入节点即可） 
	if(ro == -1){
		ro = loc;
		return;
	}
	fabs(data[loc]) < fabs(data[ro])?Insert(le[ro],loc):Insert(rig[ro],loc);
	return;
}
bool Judge(int ro,int cnt){//判断是否为red-black tree，相当于遍历整棵树 
	if(ro == -1){//到达leaf，判断当前路径上black node的个数是否与之前的相同 
		if(len == -1){
			len = cnt;
			return true;
		}
		return cnt == len;
	}
	if(data[ro] < 0){//到达red node，判断两个儿子节点是否为black node 
		if(le[ro] != -1 && data[le[ro]] < 0) return false;
		if(rig[ro] != -1 && data[rig[ro]] < 0) return false;
	}
	//递归判断左右子树 
	return Judge(le[ro],cnt + (data[ro] > 0)) && Judge(rig[ro],cnt + (data[ro] > 0));
}
int main(){
	int t,n;
	bool flag;
	scanf("%d",&t);
	while(t--){
		root = -1;len = -1;
		memset(le,-1,sizeof(le));
		memset(rig,-1,sizeof(rig));
		scanf("%d",&n);
		for(int i = 0;i < n;i++){
			scanf("%d",&data[i]);
			Insert(root,i);
		}
		flag = Judge(root,0);
		if(data[root] < 0) flag = false;//判断root是否是black node 
		puts(flag?"Yes":"No");
	}
	return 0;
}