⭐⭐⭐⭐⭐【红黑树】PAT A1135 Is It A Red-Black Tree

题目描述

知识点

BST

实现

码前思考

1. 这5个性质里，第3个和第5个我一时没看懂，还是wiki了一下才明白的。红黑树的定义里，叶子节点是NULL节点，而不是通常所说的左右孩子为空的节点是NULL节点，也就是说每个左右孩子为空的节点，实际上左右孩子都是叶子节点。然后，性质5是说每个节点到达叶子节点(NULL）节点的简单路径上经过的黑色节点总数相等。
2. 此外，我居然不知道descendant是 后辈 的意思，我这英语水平。。。
3. 扫清了上面的疑问，就很好解题了：
   判断以下几点：
   1. 根结点是否为黑色
   2. 如果一个结点是红色，它的孩子节点是否都为黑色
   3. 从任意结点到叶子结点的路径中，黑色结点的个数是否相同

代码实现

#include "bits/stdc++.h"
using namespace std;
const int maxn = 40;

struct node{
	int val;
	node* l;
	node* r;
	node(int v):val(v),l(NULL),r(NULL){}
};

int k;
int n;
int pre[maxn];
bool flag; 

//使用前序加中序进行建树
//node* create(int preL,int preR){
//	if(preL > preR){
//		return NULL;
//	}else{
//		//寻找左右子树
//		node* root = new node(pre[preL]);
//		int numLeft=0;
//		for(int i=preL+1;i<=preR;i++){
//			if(abs(pre[preL]) >= abs(pre[i])){
//				numLeft++;
//			}else{
//				break;
//			}
//		}
//		
//		//得到numLeft之后就可以划分左右子树了
//		root->l = create(preL+1,preL+numLeft);
//		root->r =  create(preL+numLeft+1,preR);
//		
//		//注意返回root 
//		return root;
//	}
//} 

void insert(node* &root,int x){
	if(root == NULL){
//		printf("%d ",x);
		root = new node(x);
	}else{
		if(abs(root->val) >= abs(x)){
			insert(root->l,x);
		}else{
			insert(root->r,x);
		}
	}
}

node* create(int n){
	node* root = NULL;
	for(int i=0;i<n;i++){
		insert(root,pre[i]);
	}
	return root;
}

//递归函数的意思代表当前根结点到叶子结点的黑色结点数
int dfs(node* root){
	if(root == NULL){
		return 0;
	}else{
		//这个结点不是空结点，那么需要进行一些判断
		//首先，这个结点是红色节点吗 
//		printf("当前访问：%d\n",root->val);		
		if(root->val < 0){
			//需要判断左右结点的值，注意判断的先后，先判空再判值！顺序反了会出错的！ 
			if((root->l==NULL||root->l->val>0) && (root->r==NULL||root->r->val>0)){
			}else{
//				printf("这里红：%d\n",root->val);
				flag = false;
			}	
		} 
		//接下来就是遍历它的左右孩子
		int num=0;
		int pl = dfs(root->l);
		int pr = dfs(root->r);
		if(pl != pr){
//			printf("这里不等：%d\n",root->val);
			flag = false;
			//随便返回哪一个都行,反正没用了 
			return pl; 
		}else{
			//如果相等的话
			if(root->val > 0){
				return pl+1;
			}else{
				return pl;
			} 
		} 
	}
} 

int main(){
	scanf("%d",&k);
	for(int i=0;i<k;i++){
		scanf("%d",&n);
		for(int j=0;j<n;j++){
			scanf("%d",&pre[j]);
		}
		
		//通过先序遍历结果进行建树
		node* root = NULL;
		root = create(n); 
//		printf("%d\n",root->val);

		//下面通过二叉树的递归来进行判断是否是红黑树
		//首先判断root是否是黑色
		if(root->val < 0){
			printf("No\n");
		}else{
			//否则进行下一步的判断
			flag = true;
			dfs(root);
			if(flag == true){
				printf("Yes\n");
			}else{
				printf("No\n");
			} 
		} 
		 
	}
	return 0;
}

码后思考

1. 所谓树的题目，万变不离DFS，所以一定要掌握DFS一棵树！！！
2. 另外，通过使用传统构建BST的函数读入二叉搜索树的先序序列就可以得到BST，我上面注释的代码复杂化了问题！！！
3. 当我们的结点不多时，我们还是主张通过insert()建树，这样逻辑没有那么绕~