AVL树旋转总结

p为旋转部分旋转后的根节点

subL为旋转部分旋转后的左孩子

subR为旋转部分旋转后的右孩子

平衡因子为右子树减左子树

1.单旋转

1.1 右旋转

流程：

1）subR指针指向平衡因子为+-2的结点，p指向subR的左孩子

2）将subR左孩子设为p的右孩子（若为空则为NULL）

3）将p的右孩子设为subR

4）将subR和p的平衡因子置为0

 

1.2 左旋转

流程：

1）subL指针指向平衡因子为+-2的结点，p指向subL的右孩子

2）将subL右孩子设为p的左孩子（若为空则为NULL）

3）将p的左孩子设为subL

4）将subL和p的平衡因子置为0

2.双旋转

2.1 左右双旋

流程：

1）subR指针指向平衡因子为+-2的结点，subL指向SubR的左孩子，p指向subL的右孩子

2）将subL右孩子设为p的左孩子（若为空则为NULL）

3）将p的左孩子设为subL

4）判断若一开始p的平衡因子为0或者-1（即p有左孩子或无孩子），则subL的平衡因子为0，否则为-1(因为如果p只有右孩子的情况下，subL一定有左孩子，旋转后只剩左孩子)

5）subR的左孩子设为p的右孩子

6）p的右孩子设为subR

7）判断若一开始p的平衡因子为0或者1（即p有右孩子或无孩子），则subR的平衡因子为0，否则为1(因为如果p只有左孩子的情况下，subL一定有右孩子，旋转后只剩右孩子)

2.2 右左双旋

流程：

1）subL指针指向平衡因子为+-2的结点，subR指向SubL的右孩子，p指向subR的左孩子

2）将subR左孩子设为p的右孩子（若为空则为NULL）

3）将p的右孩子设为subR

4）判断若一开始p的平衡因子为0或者1（即p有右孩子或无孩子），则subR的平衡因子为0，否则为1(因为如果p只有左孩子的情况下，subR一定有右孩子，旋转后只剩右孩子)

5）subL的右孩子设为p的左孩子

6）p的左孩子设为subL

7）判断若一开始p的平衡因子为0或者-1（即p有左孩子或无孩子），则subL的平衡因子为0，否则为-1(因为如果p只有右孩子的情况下，subR一定有左孩子，旋转后只剩左孩子)

记住旋转后需要break出出栈的循环，因为旋转后那一部分的平衡因子变为0，对其他部分不造成影响了。出循环后需要再判断栈是否空，不空的话，要让栈顶结点指向旋转部分的根节点。若空需要将树的根节点指向旋转部分的根节点。