平衡二叉树( AVL树 ) (zz)

本文转自：http://blog.csdn.net/lyso1/archive/2010/05/27/5627898.aspx，如有版权问题，请即使通知我删除！！！

平衡二叉树 ( AVL 树 )：

二 叉排序树的查找与树的形态密切相关，当树的形态比较均衡时查找效率最高，当树的形态偏向于某一个方向是效率迅速降低，而一颗二叉树的形态取决于数据插入的 先后顺序，如果构造一颗较均衡的二叉树比较困难。那有没有合适的方法把一颗不平衡的树调整为平衡的二叉树，答案是肯定的。这就是平衡二叉树在 1962 年由 Adelson-Velskii 和 Landis 提出的，所以又叫 AVL 树。

    其特点是：或者是一颗空树，或者是满足下列性质的二叉树：树的左子树和右子树的深度之差的绝对值不大于 1 且左右子树也满足上述性质。

平衡因子：二叉树上任一结点的左子树深度减去右子树的深度称为该结点的平衡因子，易知平衡二叉树中所有结点的因子只可能为 0 ， -1 和 1.

平 衡二叉排序树的在平衡因子绝对值等于2时开始调整到绝对值为1或0，在平衡因子绝对值为2时，二叉排序树会出现四种不同的情况的树形，因此这时需要分别单 独讨论来降低平衡因子。由于前面画的图都没有上传上来，暂时还不知道CSDN博客里怎么上传图，所以这里暂不做详解。如果要讨论四种情况，请参见源代码细 细分析： 1. #include <iostream.h> 2. #include <string.h> 3. #define NUM 10 4. typedef int KeyType; 5. class AVLTree; 6. class AVLNode 7. { 8. public: 9. KeyType key; 10. //任意一结点的左子树深度减去右子树的 11. //深度称为该节点的平衡因子. 12. int bf; //记录平衡因子 13. AVLNode *lchild; 14. AVLNode *rchild; 15. AVLNode() 16. { 17. lchild = NULL; 18. rchild = NULL; 19. bf = 0; 20. } 21. }; 22. 23. //平衡二叉排序树 24. class AVLTree 25. { 26. public: 27. AVLNode *root; 28. AVLTree() 29. { 30. root = NULL; 31. } 32. AVLNode* LL_Rotate( AVLNode *a ); //LL(顺时针)型调整 33. AVLNode* RR_Rotate( AVLNode *a ); //RR(逆时针)型调整 34. AVLNode* LR_Rotate( AVLNode *a ); //LR(先逆后顺)型调整 35. AVLNode* RL_Rotate( AVLNode *a ); //RL(先顺后逆)型调整 36. void AVLInsert( AVLNode *&pavlt, AVLNode *s ); //插入一个新结点 37. }; 38. 39. /** 40. * LL(顺时针)型调整 41. * 42. */ 43. AVLNode* AVLTree::LL_Rotate( AVLNode *a ) 44. { 45. if( a == NULL ) 46. { 47. cout << "the pointer is null!" << endl; 48. return NULL; 49. } 50. AVLNode *b; 51. b = a->lchild; //b指向a的左子树根结点 52. a->lchild = b->rchild; //b的右子树挂在a的左子树上 53. b->rchild = a; 54. a->bf = b->bf = 0; 55. return b; 56. } 57. 58. /** 59. * RR(逆时针)型调整 60. * 61. */ 62. AVLNode* AVLTree::RR_Rotate( AVLNode *a ) 63. { 64. if( a == NULL ) 65. { 66. cout << "the pointer is null!" << endl; 67. return NULL; 68. } 69. AVLNode *b; 70. b = a->rchild; 71. a->rchild = b->lchild; 72. b->lchild = a; 73. a->bf = b->bf = 0; 74. return b; 75. } 76. 77. /** 78. * LR(先逆后顺)型调整 79. * 80. */ 81. AVLNode* AVLTree::LR_Rotate( AVLNode *a ) 82. { 83. if( a == NULL ) 84. { 85. cout << "the pointer is null!" << endl; 86. return NULL; 87. } 88. AVLNode *b, *c; 89. b = a->lchild; 90. c = b->rchild; 91. a->lchild = c->rchild; 92. b->rchild = c->lchild; 93. c->lchild = b; 94. c->rchild = a; 95. 96. //调整平衡因子 97. if( c->bf == 1 ) 98. { 99. a->bf = -1; 100. b->bf = 0; 101. } 102. else if( c->bf == -1 ) 103. { 104. a->bf = 0; 105. b->bf = 1; 106. } 107. else 108. { 109. b->bf = a->bf = 0; 110. } 111. c->bf = 0; 112. return c; 113. } 114. 115. /** 116. * RL(先顺后逆)型调整 117. * 118. */ 119. AVLNode* AVLTree::RL_Rotate( AVLNode *a ) 120. { 121. if( a == NULL ) 122. { 123. cout << "the pointer is null!" << endl; 124. return NULL; 125. } 126. AVLNode *b, *c; 127. b = a->rchild; 128. c = b->lchild; 129. a->rchild = c->lchild; 130. b->lchild = c->rchild; 131. c->lchild = a; 132. c->rchild = b; 133. 134. //调整平衡因子 135. if( c->bf == 1 ) 136. { 137. a->bf = 0; 138. b->bf = -1; 139. } 140. else if( c->bf == -1 ) 141. { 142. a->bf = 1; 143. b->bf = 0; 144. } 145. else 146. { 147. a->bf = b->bf = 0; 148. } 149. c->bf = 0; 150. return c; 151. } 152. 153. /** 154. * 将结点s插入pavlt为根结点的平衡二叉排序树中 155. * 156. */ 157. void AVLTree::AVLInsert( AVLNode *&pavlt, AVLNode *s ) 158. { 159. AVLNode *f, *a, *b, *p, *q; 160. if( pavlt == NULL ) 161. { 162. pavlt = s; 163. return; 164. } 165. a = pavlt; 166. f = NULL; 167. p = pavlt; 168. q = NULL; 169. //寻找插入点位置及最小不平衡树的子树 170. while( p != NULL ) 171. { 172. if( p->key == s->key ) //AVL中已经存在关键字 173. return; 174. if( p->bf != 0 ) //寻找最小不平衡子树 175. { 176. a = p; 177. f = q; 178. } 179. q = p; 180. if( s->key < p->key ) 181. p = p->lchild; 182. else 183. p = p->rchild; 184. } 185. 186. if( s->key < q->key ) //将结点*s插入到合适的位置上去 187. q->lchild = s; 188. else 189. q->rchild = s; 190. p = a; 191. while( p != s ) //插入结点后修改相应的平衡因子 192. { 193. if( s->key < p->key ) 194. { 195. p->bf++; 196. p = p->lchild; 197. } 198. else 199. { 200. p->bf--; 201. p = p->rchild; 202. } 203. } 204. if( a->bf > -2 && a->bf < 2 ) //插入结点后没有破坏平衡树 205. return; 206. if( a->bf == 2 ) 207. { 208. b = a->lchild; 209. if( b->bf == 1 ) //结点插在*a的左孩子的左子树中 210. p = LL_Rotate( a ); //LL型调整 211. else //结点插在*a的左孩子的右子树中 212. p = LR_Rotate( a ); //LR型调整 213. } 214. else 215. { 216. b = a->rchild; 217. if( b->bf == 1 ) //结点插在*a的右孩子的左子树中 218. p = RL_Rotate( a ); //RL型调整 219. else //结点插在*a的右孩子的右子树中 220. p = RR_Rotate( a ); //RR型调整 221. } 222. if( f == NULL ) //原*a是AVL树的根 223. pavlt = p; 224. else if( f->lchild == a ) //将新子树链到原结点*a的双亲结点上 225. f->lchild = p; 226. else 227. f->rchild = p; 228. } 229. 230. /** 231. * 在VC6.0下测试成功 232. * cheneagle：2009.8.6 233. */ 234. int main( void ) 235. { 236. int a[NUM] = { 34, 18, 13, 73, 16, 52, 58, 67, 82, 76 }; 237. int i = 0; 238. AVLTree tree; 239. AVLNode pNode[NUM], *p = NULL; 240. for( i = 0; i < NUM; i++ ) 241. { 242. pNode[i].key = a[i]; 243. tree.AVLInsert( p, &pNode[i] ); 244. } 245. return 0; 246. }