1 package ren.laughing.datastructure.baseImpl; 2 3 /** 4 * 自平衡二叉查找树AVL 继承二叉查找树 5 * 6 * @author Laughing_Lz 7 * @time 2016年4月20日 8 */ 9 public class AVLTree extends BSTree { 10 /** 11 * 统一平衡方法 旋转操作 12 * 13 * @param z 14 * z是失衡的结点 15 * @return 返回平衡后子树的根结点 16 */ 17 private BinTreeNode rotate(BinTreeNode z) { 18 BinTreeNode y = heightSubT(z);// 取y为z更高的孩子 19 BinTreeNode x = heightSubT(y);// 取x为y更高的孩子 20 boolean isLeft =z.isLChild();//标记z是否是左孩子 21 BinTreeNode p = z.getParent();//取z的父结点 22 BinTreeNode a,b,c; 23 BinTreeNode t0,t1,t2,t3; 24 //以下分四种情况重命名 25 if(y.isLChild()){//1如果y是左孩子 26 c = z; 27 t3 = z.getRChild(); 28 if(x.isLChild()){//1.1如果x是左孩子(左左失衡) 29 a = x; 30 b = y; 31 t0 = x.getLChild(); 32 t1 = x.getRChild(); 33 t2 = y.getRChild(); 34 }else{//1.2如果x是右孩子(左右失衡) 35 b = x; 36 a = y; 37 t1 = x.getLChild(); 38 t2 = x.getRChild(); 39 t0 = y.getLChild(); 40 } 41 }else{//2如果y是右孩子 42 a = z; 43 t0 = z.getLChild(); 44 if(x.isRChild()){//2.1如果x是右孩子(右右失衡) 45 c = x; 46 b = y; 47 t2 = x.getLChild(); 48 t3 = x.getRChild(); 49 t1 = y.getLChild(); 50 }else{//2.2如果x是左孩子(右左失衡) 51 b = x; 52 c = y; 53 t1 = x.getLChild(); 54 t2 =x.getRChild(); 55 t3 = y.getRChild(); 56 } 57 } 58 // x.server();//疑问,已经赋值给a b c 还有何用?★ 59 // y.server(); 60 // z.server(); 61 a.server();//断开与父结点的连接 62 b.server(); 63 c.server(); 64 if(t0 != null){ 65 t0.server(); 66 } 67 if(t1 != null){ 68 t1.server(); 69 } 70 if(t2 != null){ 71 t2.server(); 72 } 73 if(t3 != null){ 74 t3.server(); 75 } 76 b.setLChild(a);//重新设置左右孩子,平衡 77 b.setRChild(c); 78 a.setLChild(t0); 79 a.setRChild(t1); 80 c.setLChild(t2); 81 c.setRChild(t3); 82 if(isLeft){ 83 p.setLChild(b); 84 }else{ 85 p.setRChild(b); 86 } 87 return b;//返回平衡后子树的根 88 } 89 90 /** 91 * 获取结点v较高的子树 92 * 93 * @param v 94 * @return 95 */ 96 private BinTreeNode heightSubT(BinTreeNode v) { 97 if (v == null) { 98 return null; 99 } 100 int lH = -1;// 若没有左子树,默认左子树高度为-1 101 if (v.hasLChild()) { 102 lH = v.getLChild().getHeight(); 103 } 104 int rH = -1;// 若没有右子树,默认右子树高度为-1 105 if (v.hasRChild()) { 106 rH = v.getRChild().getHeight(); 107 } 108 if (lH > rH) { 109 return v.getLChild(); 110 } else if (lH < rH) { 111 return v.getRChild(); 112 } else {// 若两边子树等高 113 if (v.isLChild()) { 114 return v.getLChild(); 115 } else { 116 return v.getRChild(); 117 } 118 } 119 } 120 /** 121 * 重写BSTree的insert方法 122 * insert之后要重新平衡 123 */ 124 @Override 125 public void insert(Object obj) { 126 super.insert(obj); 127 root = reBalance(startBN);//重新平衡树 128 } 129 /** 130 * 重新平衡AVL树 131 * @param startBN 132 * @return 返回平衡后的AVL树根结点 133 */ 134 private BinTreeNode reBalance(BinTreeNode startBN) { 135 if(startBN == null){ 136 return root; 137 } 138 BinTreeNode c = startBN; 139 while(startBN != null){//从startBN开始,向上逐一检查 z 的祖先★ 140 if(!isBalance(startBN)){//若startBN失衡,则旋转使之重新平衡★ 141 startBN = rotate(startBN); 142 } 143 c = startBN; 144 startBN = startBN.getParent();//继续检查其父亲 145 } 146 return c; 147 } 148 /** 149 * 判断该结点是否平衡 150 * @param startBN 151 * @return 152 */ 153 private boolean isBalance(BinTreeNode startBN) { 154 if(startBN == null){ 155 return true; 156 } 157 int lH = startBN.hasLChild()?startBN.getLChild().getHeight():-1; 158 int rH = startBN.hasRChild()?startBN.getRChild().getHeight():-1; 159 return (Math.abs(lH-rH) <= 1); 160 } 161 /** 162 * 重写BSTree的remove方法 163 */ 164 @Override 165 public Object remove(Object ele) { 166 Object obj = super.remove(ele); 167 root = reBalance(startBN);//重新平衡AVL树 168 return obj; 169 } 170 }
时间: 2024-10-31 15:12:52