问题描述:给定一颗树的先序,判断是否为红黑树。
解题思路:满足BST,直接插入建树,判断需每个节点都判断,按理来说,只判断根节点到所有叶节点的黑节点个数是否相等即可呀???????
AC代码:
#include<iostream>
#include<unordered_map>
#include<algorithm>
#include<cmath>
using namespace std;
typedef struct node{
int left,right;
node(){}//需要这个,不然会出错!!!!!
node(int l,int r):left(l),right(r){}
}node;
int M=0;
unordered_map<int,node>arr;
void BSTinsert(int &v,int x)//建树,v相当于指针,
{
if(!v){v=x;return;}
if(abs(v)<abs(x)){//往右子树插
BSTinsert(arr[v].right,x);
}else{//左子树插
BSTinsert(arr[v].left,x);
}
}
/*void pre(int v)
{
if(!v)return;
printf("%d ",v);
pre(arr[v].left);
pre(arr[v].right);
}*/
/*void judge(int v,bool &f,int cnt)//测试点3错误,没有检查其他节点,只检测了根节点。
{
if(!v){//叶节点
if(!M)M=cnt;
else if(M!=cnt)f=1;
return;
}else if(v>0){//黑节点
++cnt;
}else if(arr[v].right<0||arr[v].left<0){//红节点
f=1;return;
}
judge(arr[v].left,f,cnt);//左子树
if(!f)judge(arr[v].right,f,cnt);//右子树
}*/
int judge(int v,bool &f)
{
if(!v||f)return 1;//叶节点
int l=judge(arr[v].left,f);
int r=judge(arr[v].right,f);
if(l!=r)f=1;//左右黑节点个数不一样
if(v<0&&(arr[v].right<0||arr[v].left<0))f=1;//红节点
if(v>0)++l;//黑节点
return l;
}
int main()
{
freopen("test.txt","r",stdin);
int K,N,root,a;
bool flag=0;
scanf("%d",&K);
while(K--){
arr.clear();flag=0;
scanf("%d %d",&N,&root);
if(root<0)flag=1;//根为红
arr[root]=node(0,0);
while(--N){
scanf("%d",&a);
if(flag)continue;
arr[a]=node(0,0);
BSTinsert(root,a);
}
// pre(root);printf("\n");
if(!flag)judge(root,flag);
if(!flag)printf("Yes\n");
else printf("No\n");
}
return 0;
}