AVL树的插入实现

1、AVL树的概念

2.AVL树实现

首先需要有节点然后需要组织链式结构，所以定义下面两个类。

节点

struct AVLTreeNode
{
	AVLTreeNode* _left;
	AVLTreeNode* _right;
	AVLTreeNode* _parent;

	pair<k, v> _kv;
	int _bf;//存储平衡因子方便旋转

	AVLTreeNode(const pair<k, v>& kv)
		:_left(nullptr),
		_right(nullptr),
		_parent(nullptr),
		_kv(kv),
		_bf(0)
	{}
};

AVLTree

template<class k, class v>
class AVLTree
{
	typedef AVLTreeNode<k, v> Node;
    private:
    Node* _root;
}

AVL树的左右子树高度差不超过1，左右子树也为AVL树，当AVL树不平衡时需要将对不平衡的部分进行旋转变得平衡，下面介绍怎么旋转。

2.1 旋转

2.1.1 左单璇

当我们插入一个新的节点单纯的右边高时通过调整平衡因子不正常的节点，即将该节点进行左单璇，下面给出图解。

经过左旋变为下面的图。

可以看到parent接管了subR的左孩子，而parent 变为了subR的右孩子，到这好像是链接完成了，但是我们这有可能只是作为AVL的一部分而已，有可能如下图所示，所以我们还需要将suR与parent的父亲节点进行链接作为左子树或者右子树（需要进行判断确定链接为左子树还是右子树）。所以我们需要做的就是更改三处链接关系，由于我们定义了parent节点，所以我们一共需要更改6个指针（最多），因为有可能h为0的情况下，更改的指针就没有6个。最后不要忘记将平衡因子修正一下。

下面给出代码演示

	void RotateL(Node* parent)
	{
		Node* ppnode = parent->_parent;

		Node* subR = parent->_right;
		Node* subRL = subR->_left;

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

		subR->_left = parent;
		parent->_parent = subR;

		if (parent == _root)
		{
			_root = subR;

		}
		else
		{
			if (ppnode->_right == parent)
			{
				ppnode->_right = subR;
			}
			else if (ppnode->_left == parent)
			{
				ppnode->_left = subR;
			}

		}
		subR->_parent = ppnode;
		subR->_bf = parent->_bf = 0;
	
	}

右单旋和左单璇是一样逻辑这里就不多说了，直接给出代码。

	void RotateR(Node* parent)
	{
		Node* ppnode = parent->_parent;

		Node* subL = parent->_left;
		Node* subLR = subL->_right;

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

		subL->_right = parent;
		parent->_parent = subL;

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

	}

2.1.2  右左双旋

当我们新插入的节点不是单纯的右边高时，即插入在下图中的b或者c处时

首先需要对subR进行右单旋，使得该树变为单纯的右边高（这样就将问题转化得更简单了），然后再做左单璇，通过两次旋转，使得AVL树平衡。最后调整平衡因子。

需要注意的是有特殊情况需要进行处理，即当h为0时b节点为空，插入b节点引发双旋。

三种情况的平衡因子不一样需要分别调整，如何区分上面介绍的三种情况呢，可以观察到上面三种情况的subRL的平衡因子分别为1，-1，0所以我们可以通过它的平衡因子来分类，分别调整平衡因子。下面给出代码实现。

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

		int bf = subRL->_bf;

		RotateR(parent->_right);
		RotateL(parent);
		//调整平衡因子
		subRL->_bf = 0;
		if (bf == 0)
		{
			subR->_bf = 0;
			parent->_bf = 0;
		}
		else if (bf == 1)
		{
			subR->_bf = 0;
			parent->_bf = -1;

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

同样左右双旋这里直接给出代码

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

		int bf = subLR->_bf;
		//LR左右双旋
		RotateL(parent->_left);
		RotateR(parent);
		//调整平衡因子
		subLR->_bf = 0;
		if (bf == 0)
		{
			subL->_bf = 0;
			parent->_bf = 0;
		}
		else if (bf == 1)
		{
			subL->_bf = -1;
			parent->_bf = 0;
		}
		else if (bf == -1)
		{
			subL->_bf = 0;
			parent->_bf = 1;
		}
		else
		{
			assert(false);
		}
	}