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

题目地址：Is It A Red-Black Tree

思路：

1. 根据二叉搜索树的特点，其中序遍历的结果就是从小到大的结果，再结合给出的前序遍历就可以构建出二叉树。
2. 从根节点开始搜索，若发现红色的节点，检查左，右儿子是否都是黑色节点，若存在红色节点则为No。
3. 从根节点开始搜索，检查每个节点的“深度”是否合法。
4. 特判根节点的颜色。

代码实现：

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
typedef struct Tnode *p;
const int N=55;
struct Tnode{
	p ls,rs;
	int data;
};
p root;
int n,preindex,preorder[N],inorder[N],a[N],b[N];
bool vis[10005];
void init(){
	root=NULL;
	memset(vis,0,sizeof(vis));
	preindex=0;
}
int find(int begin,int end,int val){
	for(int i=begin;i<=end;i++){
		if(inorder[i]==val)	return i;
	}
	return -1;
}
p buildtreechild(int begin,int end){
	if(begin>end)	return NULL;
	p root=new Tnode;
	root->data=preorder[preindex];
	int index=find(begin,end,preorder[preindex]);
	if(index==-1)	return NULL;
	preindex++;
	root->ls=buildtreechild(begin,index-1);
	root->rs=buildtreechild(index+1,end);
	return root;
}
p buildtree(){
	if(preorder==NULL||inorder==NULL)	return NULL;
	return buildtreechild(0,n-1);
}
bool check1(p u){
	if(u==NULL)	return true;
	if(vis[u->data]){
		if(u->ls&&vis[u->ls->data])	return false;
		if(u->rs&&vis[u->rs->data])	return false;
	}
	return check1(u->ls)&&check1(u->rs);
}
int depth(p u){
	if(u==NULL)	return 0;
	int left=depth(u->ls);
	int right=depth(u->rs);
	if(vis[u->data])	return max(left,right);
	return max(left,right)+1;
}
bool check2(p u){
	if(u==NULL)	return true;
	int left=depth(u->ls);
	int right=depth(u->rs);
	if(left!=right)	return false;
	return check2(u->ls)&&check2(u->rs);
}
int main(){
	int t;
	scanf("%d",&t);
	while(t--){
		init();
		scanf("%d",&n);
		for(int i=0;i<n;i++){
			scanf("%d",&a[i]);
			if(a[i]<0)	preorder[i]=-a[i],vis[abs(a[i])]=true;
			else	preorder[i]=a[i];
			b[i]=abs(a[i]); 
		}
		sort(b,b+n);
		for(int i=0;i<n;i++)	inorder[i]=b[i];
		root=buildtree();
		if(a[0]<0||!check1(root)||!check2(root))	puts("No");
		else	puts("Yes");
	}
	return 0;
}