AVL树与红黑树的实现

• 我最近开了几个专栏，诚信互三！
  ====> |||《算法专栏》：：刷题教程来自网站《代码随想录》。|||
  ====> |||《C++专栏》：：记录我学习C++的经历，看完你一定会有收获。|||
  ====> |||《Linux专栏》：：记录我学习Linux的经历，看完你一定会有收获。|||
  ====> |||《C#专栏》：：记录我复习C#的经历，深度理解，查漏补缺，不定期更新。|||
  ====> |||《计算机网络专栏》：：记录我学习计算机网络，看完你一定会有收获。|||

二叉搜索树的原理

二叉搜索树（BST），是一种树的特殊结构，该树的左结点比根结点小，右结点比根结点大。

二叉搜索树之所以叫做二叉搜索树，是因为其搜索方便，二叉搜索树的搜索类似于二分法，每次都能去除树的一半的结点。
在理想状态下，二叉搜索树查找某一结点的时间复杂度为O(logn)，但是二叉搜索树在最差的情况下会退化为链表，这样查找的时间复杂度就会下降为O(n)。

二叉搜索树有一个重要特性，二叉搜索树的中序遍历是有序的。

AVL树的实现

AVL树是为了解决BST最坏情况下的一种新的数据结构，其将每个结点中存储一个平衡因子(bf)，来保证BST不会退化为最坏情况。

平衡因子(bf) = 右子树高度 - 左子树高度。

在AVL树中，若bf > 1 || bf < -1则需要对该结点进行旋转操作，来保证bf一直处于[1,-1]之间。
此时需要旋转的情况有4种
一.parent.bf == 2 && cur.bf == 1

此种情况需要进行左单旋操作，保持平衡。

void RotateL(Node* parent)
{
	Node* subR = parent->_right;
	Node* subRL = subR->_left;

	parent->_right = subRL;
	subR->_left = parent;
	if (subRL)
		subRL->_parent = parent;

	Node* ppnode = parent->_parent;
	parent->_parent = subR;

	if (parent == _root)
	{
		_root = subR;
		_root->_parent = nullptr;
	}
	else
	{
		if (ppnode->_right == parent)
		{
			ppnode->_right = subR;
		}
		else
		{
			ppnode->_left = subR;
		}
		subR->_parent = ppnode;
	}
	subR->_bf = 0;
	parent->_bf = 0;


	return;

}

二.parent.bf == 2 && cur.bf == -1

此时需要右左双旋操作，保持平衡，且60结点也有两种情况，有可能新结点插入b下，还可能插入c下，这会导致某些结点平衡因子变化。
(1).先要对90进行右单旋

(2).在对30进行左单旋。

void RotateRL(Node* parent)
{
	Node* subR = parent->_right;
	Node* subRL = subR->_left;


	int bf = subRL->_bf;
	RotateR(subR); // - 进行右单旋
	RotateL(parent);// - 进行左单旋

	if (bf == -1) // - 新结点插入b下
	{
		subRL->_bf = 0;
		subR->_bf = 1;
		parent->_bf = 0;
	}
	else if (bf == 1)// - 新结点插入c下
	{
		parent->_bf = -1;
		subR->_bf = 0;
		subRL->_bf = 0;
	}
	else
	{
		assert(false);
	}
	return;


}

三.parent.bf == -2 && cur.bf == 1

此时需要进行左右双旋，先对30进行左旋，在对90进行右旋。
(1).先对30进行左旋

(2).在对90进行右旋

void RotateLR(Node* parent)
	{
		Node* subL = parent->_left;
		Node* subLR = subL->_right;

		int bf = subLR->_bf;
		RotateL(parent->_left);
		RotateR(parent);

		if (bf == -1)
		{
			subLR->_bf = 0;
			subL->_bf = 0;
			parent->_bf = 1;
		}
		else if (bf == 1)
		{
			subLR->_bf = 0;
			subL->_bf = -1;
			parent->_bf = 0;
		}
		else if (bf == 0)
		{
			subLR->_bf = 0;
			subL->_bf = 0;
			parent->_bf = 0;
		}
		else
		{
			assert(false);
		}
	}

四.parent.bf == -2 && cur.bf == -1

此时需要对60进行右旋。

void RotateR(Node* parent) 
	{
		Node* subL = parent->_left;
		Node* subLR = subL->_right;

		parent->_left = subLR;
		if (subLR)
			subLR->_parent = parent;

		subL->_right = parent;

		Node* ppnode = parent->_parent;
		parent->_parent = subL;

		if (parent == _root)
		{
			_root = subL;
			subL->_parent = nullptr;
		}
		else
		{
			if (ppnode->_left == parent)
			{
				ppnode->_left = subL;
			}
			else
			{
				ppnode->_right = subL;
			}
			subL->_parent = ppnode;
		}

		subL->_bf = 0;
		parent->_bf = 0;
	}

红黑树的实现

红黑也是一种平衡方法，不同于AVL给每个结点定义平衡因子，而是给每个结点定义一个颜色变量，给每个结点编色，红黑树的最长路径不超过最短路径的二倍。

要实现红黑树，要满足以下三个条件。
1.红黑树的根结点必须是黑色的。
2.红黑树的红色结点不能连续出现。
3.红黑树的最短路径不小于最长路径的二分之一。
满足以上三个条件，该树就是红黑树。

在我们进行插入操作时，我们要插入的结点是红色结点，在插入了红色结点后，会有以下情况。
插入结点后，结点的父亲结点和祖父结点的颜色一定是已知的（父亲结点是红色，祖父结点是黑色）， 只有叔叔结点的颜色会有区别。
1.叔叔结点存在并且是红色。

此时不需要旋转，只需要将p结点和u结点变为黑色，吧g结点变为红色，随后还要以g结点为插入结点，继续向上判断，将该树镜像，也需要这样操作。
2.叔叔存在且为黑色/不存在，cur在p的左侧。

此时需要进行旋转操作，若cur的父亲是g的左子树，对g进行右旋操作，若cur的父亲是g的右子树，对g进行左旋，将p的颜色置为黑，g结点置为红。
3.叔叔存在且为黑/不存在，cur在p的右侧

此时需要进行旋转操作，p为g的左孩子，cur为p的右孩子，则针对p做左单旋转；相反，
p为g的右孩子，cur为p的左孩子，则针对p做右单旋转，此时该情况就变为了第二种情况，只需要根据第二种情况进行旋转就可以了。