标签：pre PAT cur puts Level int Tree dfs flag

判断是否为红黑树，主要判断子树到叶子节点的路径上黑色的数量是否相等
还需要判断给定的前序遍历是否是二叉搜索树的合法表示，虽然题目没说（pat特色）

#include<bits/stdc++.h>
using namespace std;

typedef long long LL;

const int N = 40;

int pre[N],in[N],idx;

int dfs(int cur,int n,int color,bool &flag){//0红色，1黑色，初始-1 
	if(n<=0) return 0;
	if(color==-1 && pre[cur]<0){
		flag=false;
		return 0;
	}
	if(color==0 && pre[cur]<0){
		flag=false;
		return 0;
	}
	int pos=cur+n;
	for(int i=cur+1;i<cur+n;i++)
		if(abs(pre[i])>=abs(pre[cur])){
			pos=i;
			break;
		}
	int clr;
	if(pre[cur]<0) clr=0;
	else clr=1; 
	int cnt1=dfs(cur+1,pos-cur-1,clr,flag);
	if(abs(pre[cur])!=in[++idx]) flag=false;
	int cnt2=dfs(pos,n-(pos-cur-1)-1,clr,flag);
	if(cnt1!=cnt2){
		flag=false;
		return 0;
	}
	if(clr) return cnt1+1;
	else return cnt1;
} 

int main(){
	int T;
	cin>>T;
	while(T--){
	    idx=0;
		int n;
		cin>>n;
		bool flag=true;
		for(int i=1;i<=n;i++) cin>>pre[i],in[i]=abs(pre[i]);
		sort(in+1,in+n+1);
		dfs(1,n,-1,flag);
		if(flag) puts("Yes");
		else puts("No");
	}
	return 0;
}

标签：pre,PAT,cur,puts,Level,int,Tree,dfs,flag
来源： https://www.cnblogs.com/xhy666/p/16340115.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。