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,对其他部分不造成影响了。出循环后需要再判断栈是否空,不空的话,要让栈顶结点指向旋转部分的根节点。若空需要将树的根节点指向旋转部分的根节点。