PAT甲级1135

做出来了，但感觉心好痛。我发现我真是太懒惰了，以为简简单单记个模板，就万事大吉了。

没想到自己的思维被模板固化，就不会根据实际情况作出相应的改变了。模板当然要记忆，

但不可偷懒而导致思维僵化。引以为戒

dfs为判断过程，是在求树高的基础上，增加一个参数，判断是否存在前一个为红色节点并且当前节点是红色的情况

回溯判断当前节点的左右子树的黑高是否相等

#include<cstdio>
#include<algorithm>
#include<cmath>
//#define debug
using namespace std;
const int maxn=40;
struct node{
	int data;
	int lc;
	int rc;
	node(){
		data=0;
		lc=rc=-1;
	}
};
bool cmp(int a,int b)
{
	return abs(a)<abs(b);
}
int n;
int ptr=0;
node Tree[maxn];
int pre[maxn];
int in[maxn];
void create(int &root,int preL,int preR,int inL,int inR)
{
	if(preL>preR)return;
	if(preL>=n)return;
	int roote=pre[preL];
	root=ptr++;
	Tree[root].data=roote;
	int u=0;
	for(u=inL;u<=inR;u++){
		if(in[u]==roote)break;
	}
	int numLeft=u-inL;
	create(Tree[root].lc,preL+1,preL+numLeft,inL,u-1);
	create(Tree[root].rc,preL+numLeft+1,preR,u+1,inR);
}
bool IsRBt=true;
int dfs(int root,int pren)
{
	if(IsRBt==false)return -1;
	if(root==-1)return 1;
	if(pren<0&&Tree[root].data<0){
		IsRBt=false;
		return -1;
	}
	int lh=dfs(Tree[root].lc,Tree[root].data);
	int rh=dfs(Tree[root].rc,Tree[root].data);
	if(rh!=lh){
		IsRBt=false;
		return -1;
	}
	return Tree[root].data>0?max(lh,rh)+1:max(lh,rh);
}
int main()
{
#ifdef debug
	freopen("in.txt","r",stdin);
#endif
	int t;
	node s;
	scanf("%d",&t);
	for(int i=0;i<t;i++){
		scanf("%d",&n);
		fill(Tree,Tree+maxn,s);
		ptr=0;
		IsRBt=true;
		for(int i=0;i<n;i++){
			scanf("%d",&pre[i]);
			in[i]=pre[i];
		}
		sort(in,in+n,cmp);
		int root;
		create(root,0,n-1,0,n-1);
		if(Tree[root].data<0)IsRBt=false;
		else
			dfs(root,Tree[root].data);
		if(IsRBt)printf("Yes\n");
		else printf("No\n");
	}
#ifdef debug
	getchar();
#endif
}