AVL平衡二叉树图解

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76

//  AVLTree插入算法 template < class  K,  class  V> bool  AVLTree<K,V>::Insert( const  K& key,  const  V& value) {      //1.空树      if  (_root == NULL)      {          _root =  new  AVLTreeNode<K, V>(key, value);          return  true ;      }            //2.AVL树不为NULL      AVLTreeNode<K, V>* parent = NULL;      AVLTreeNode<K, V>* cur = _root;      //找到数据插入位置      while  (cur)      {          if  (cur->_key < key)          {              parent = cur;              cur = cur->_right;          }          else     if  (cur->_key > key)          {              parent = cur;              cur = cur->_left;          }          else          {              return  false ;          }      }      //插入数据          cur =  new  AVLTreeNode<K, V>(key, value);          cur->_parent = parent;          if  (parent->_key > key)              parent->_left = cur;          else              parent->_right = cur;            while  (parent)          {              //更新平衡因子              if  (cur == parent->_left)                  parent->_bf--;              else  if  (cur == parent->_right)                  parent->_bf++;                //检验平衡因子是否合法              if  (parent->_bf == 0)                  break ;              else  if  (parent->_bf == -1 || parent->_bf == 1)              {   // 回溯上升 更新祖父节点的平衡因子并检验合法性                  cur = parent;                  parent = cur->_parent;              }              else   //  2 -2 平衡因子不合法 需要进行旋转 降低高度              {                  if  (parent->_bf == 2)                  {                      if  (cur->_bf == 1)                          _RotateL(parent);                      else                          _RotateRL(parent);                  }                  else  if  (parent->_bf == -2)                  {                      if  (cur->_bf == -1)                          _RotateR(parent);                      else                          _RotateLR(parent);                  }                  break ;              }          } }

　　　

---

左旋的两种情况：

1.parent有两个孩子：没有插入节点c之前处于平衡状态，插入c之后，平衡被破坏，向上回溯检验祖父节点的平衡因子，当其bf=2 时，以此节点为轴进行左旋

2.parent有一个孩子：没有插入节点a之前处于平衡状态，插入节点a之后，parent节点的平衡因子bf=2不满足AVL树的性质，要以parent为轴进行左旋

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33

//左旋 template < class  K,  class  V> void  AVLTree<K, V>::_RotateL(AVLTreeNode<K, V>*&  parent) {      AVLTreeNode<K, V>* subR = parent->_right;      AVLTreeNode<K, V>* subRL = subR->_left;      AVLTreeNode<K, V>* ppNode = parent->_parent;       //标记祖先节点        //1.构建parent子树 链接parent和subRL      parent->_right = subRL;      if  (subRL) subRL->_parent = parent;      //2.构建subR子树 链接parent和subR      subR->_left = parent;      parent->_parent = subR;      //3.链接祖先节点和subR节点      subR->_parent = ppNode;      if  (ppNode== NULL)      { //如果祖先节点为NULL，说明目前的根节点为subR          _root = subR;      }      else      {   //将祖先节点和subR节点链接起来          if  (parent == ppNode->_left)              ppNode->_left = subR;          else              ppNode->_right = subR;      }      //4.重置平衡因子      parent->_bf = 0;      subR->_bf = 0;      //5.更新subR为当前父节点      parent = subR; }

---

右旋的两种情况：

1. parent既有左孩子又有右孩子：插入c之前处于平衡态，插入c之后parent的平衡因子变为-2，这时要以parent为轴进行旋转

 

2. parent只有一个孩子：插入a之前处于平衡状态，插入之后subL与parent的平衡因子被改变，需要以parent为轴进行旋转

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33

///右旋 template < class  K,  class  V> void  AVLTree<K, V>::_RotateR(AVLTreeNode<K, V>*&  parent) {      AVLTreeNode<K, V>* subL = parent->_left;      AVLTreeNode<K, V>* subLR = subL->_right;      AVLTreeNode<K, V>* ppNode = parent->_parent;       //标记祖先节点      //1.构建parent子树 将parent和subLR链接起来      parent->_left = subLR;      if  (subLR) subLR->_parent = parent;      //2.构建subL子树 将subL与parent链接起来      subL->_right = parent;      parent->_parent = subL;      //3.将祖先节点与sunL链接起来      if  (ppNode == NULL)      {   //如果祖先为NULL，说明当前subL节点为根节点          subL->_parent = NULL;          _root = subL;      }      else      {          subL->_parent = ppNode;          if  (ppNode->_left == parent)              ppNode->_left = subL;          else  if  (ppNode->_right == parent)              ppNode->_right = subL;      }      //4.重置平衡因子      parent->_bf = 0;      subL->_bf = 0;      //5.更新subL为当前父节点      parent = subL; }

---

 左右双旋：

1. parent只有一个孩子：在插入节点sunLR之前，AVL树处于平衡状态，左右子树高度差的绝对值不超过1。

　　由于插入了节点subLR导致grandfather的平衡因子变为-2，平衡树失衡，所以需要利用旋转来降低高度！

• 首先以subL为轴，将subLR向上提（左旋），将grandfather、parent和subL旋转至一条直线上；
• 再以parent为轴将之前的subLR向上提（右旋），左树的高度降1，grandfather的平衡因子加1后变为-1，恢复平衡状态。
• 双旋完成后将parent、subL的平衡因子置为0即可，左右双旋也就完成啦！

2. parent有两个孩子：没有插入subRL或subRR之前的AVL树一定是处于平衡状态的，并且满足AVL树的性质。

　　正是由于插入了节点subRL或者subRR，导致其祖先节点的平衡因子被改变，grandfather的平衡因子变为-2，平衡态比打破，需要进行旋转来降低高度！

• 首先parent为轴将subR节点往上提至原parent的位置（左旋），将grandfather、parent 和 subR旋至一条直线上；
• 再以grandfather为轴将subR往上提至grandfather的位置（右旋），此时以subR为根的左右子树的高度相同，恢复了平衡态！

parent有两个孩子时，要看插入的节点是subR的右孩子还是左孩子，双旋后对平衡因子的修改分两种情况：

• subR的平衡因子为1，即subR有右孩子无左孩子（有subRR但无subRL），双旋之后将grandfather的平衡因子置为0，将parent的平衡因子置为-1；
• subR的平衡因子为-1，即subR有左孩子无右孩子（有subRL但无subRR），双旋之后将grandfather的平衡因子置为1，将parent的平衡因子置为0；

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29

//左右双旋 template < class  K,  class  V> void  AVLTree<K, V>::_RotateLR(AVLTreeNode<K, V>*&  parent) {      AVLTreeNode<K, V>* pNode = parent;      AVLTreeNode<K, V>* subL = parent->_left;      AVLTreeNode<K, V>* subLR = subL->_right;      int  bf = subLR->_bf;        _RotateL(parent->_left);      _RotateR(parent);            if  (bf == 1)      {          pNode->_bf = 0;          subL->_bf = -1;      }      else  if  (bf == -1)      {          pNode->_bf = 1;          subL->_bf = 0;      }      else      {          pNode->_bf = 0;          subL->_bf = 0;      }   }

---

　右左双旋：

1. parent只有一个孩子：由于节点subRL的插入破坏了AVL树的平衡，parent的平衡因子变为2，需要利用旋转来降低高度！

• 首先，以subR为轴，将subRL提上去（右旋），保证parent、subR 和 subRL在一条直线上；
• 以parent为轴，将上一步标记为subRL的节点向上升（左旋），这样达到了降低高度的目的；
• 双旋之后，parent和subR的平衡因子都要置为0

 

2.parent有两个孩子：没有插入subLL或者subLR之前的AVL树一定是处于平衡状态的，并且满足AVL树的性质。

　　正是由于插入了节点subLL或者subLR，导致其祖先节点的平衡因子被改变，grandfather的平衡因子变为2，平衡态比打破，需要进行旋转来降低高度！

• 首先parent为轴将subL节点往上提至原parent的位置（右旋），将grandfather、parent 和 subL旋至一条直线上；
• 再以grandfather为轴将subL往上提至grandfather的位置（左旋），此时以subL为根的左右子树的高度相同，恢复了平衡态！

parent有两个孩子时，要看插入的节点是subL的右孩子还是左孩子，双旋后对平衡因子的修改分两种情况：

• subL的平衡因子为1，即subL有右孩子无左孩子（有subLR但无subLL），双旋之后将grandfather的平衡因子置为-1，将parent的平衡因子置为0；
• subL的平衡因子为-1，即subL有左孩子无右孩子（有subLL但无subLR），双旋之后将grandfather的平衡因子置为0，将parent的平衡因子置为1；　

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28

//右左双旋 template < class  K,  class  V> void  AVLTree<K, V>::_RotateRL(AVLTreeNode<K, V>*&  parent) {      AVLTreeNode<K, V>* pNode = parent;      AVLTreeNode<K, V>* subR= parent->_right;      AVLTreeNode<K, V>* subRL = subR->_left;      int  bf = subRL->_bf;        _RotateR(parent->_right);      _RotateL(parent);        if  (bf == 1)      {          pNode->_bf = 0;          subR->_bf = -1;      }      else  if  (bf == -1)      {          pNode->_bf = 1;          subR->_bf = 0;      }      else      {          pNode->_bf = 0;          subR->_bf = 0;