程序博客网 > 银行卡被网络盗刷

AVL树[非常详细有用]

来源：互联网 发布：银行卡被网络盗刷 编辑：程序博客网 时间：2024/05/22 04:32

 

定义：一棵空二叉树是AVL树，如果T是非空二叉树，TL和TR分别是其左子树和右子树，

则当且仅当TL和TR都为AVL树且|HL-HR|<=1时，T是AVL树。

由定义知道一个AVL树的任何节点的左右子树的高度之差不超过1，这是AVL树最基本的特征。

AVL树的高度：（固定节点数计算最大高度）

记N_h为一棵高度为h的AVL树具有的最小节点数，则最坏情况是它的左右子树的高度不等，

一个是N_(h-1)和N_(h-2)，从而得到

N_h=N_(h-1)+N_(h-2)+1            N_0=0,N_1=1

这类似于Fibonacci数列：F_n=F_(n-1)+F_(n-2),(F_0=0,F_1=1)

而F_h=（1+sqrt(5)）^h/sqrt(5)

从而高度h和节点数是对数关系，因此 h=O(log(N_h))

由此容易知道在不考虑恢复AVL树的前提下，它的插入，删除和查找的工作量不超过O(n)。

AVL树节点的平衡因子：

AVL树节点的平衡因子定义为其左子树的高度减去右子树的高度，我们可以在插入和删除操作的时候更新平衡因子。

一棵AVL树的各节点平衡因子为1，-1, 0

树的旋转：

在介绍插入和删除操作之前首先介绍树的旋转操作

树的旋转操作是为了改变树的结构，使其达到平衡。旋转总共分为左旋和右旋两类

左旋

如图为树的以B为轴左旋示意图，从右到左是以A为轴右旋，树的旋转操作要特别注意儿子节点是否为空

顶层节点是否为根节点。

AVL树的插入操作：

插入操作只会改变被插入节点到根节点路径上的节点的平衡因子。

因为在插入之前树是AVL树，因此各个节点的平衡因子是1，-1或0

一、如果路径上节点平衡因子是0，则插入后不会打破这个节点的平衡性。

二、如果路径上的节点的平衡因子是1或-1，则可能会打破平衡性，在这种情况下如果此节点的新的平衡

因子是0，则刚好将其补的更加平衡，平衡性未打破；否则平衡因子变成2或-2，则需要进行调整。

三、我们在对树进行调整后恢复子树相对于插入前的高度，不改变子子树的平衡因子。

由以上三点：只需向根部查找从插入点开始第一个平衡因子不为0的节点，令该节点为A

更新步骤：

(1). 若Ａ不存在，说明插入节点不影响平衡性，则自下而上更新平衡因子直到根部即可。

(2). 若bf(A)=1且插入节点在A地右子树中，或者bf(A)=-1且插入节点在A的左子树中，则自下而上更新

平衡因子直到A。（注：A的平衡因子变为0）

(3). 若A的平衡性被打破，首先依然自下而上更新平衡因子，在按照下面进行分类：LL, LR, RR, RL

原bf(A)=1, 新的bf(A)=2, LL, LR,

原bf(A)=-1, 新的bf(A)=-2, RR, RL

如图为RR的情况，LL情况类似，旋转后只需再次改变两个节点的平衡因子

RR

如下图为RL的情况，LR类似，旋转后需要再次改变三个节点的平衡因子

RL

至此，插入操作已经完成，可见最多多做两次旋转操作调整树的结构使之平衡。

AVL树的删除操作：

删除操作和插入操作一样，也是先直接删除二叉树节点，然后在更新平衡因子，

调整AVL树使之平衡。

这里所指的删除的节点是实际被删除的节点，删除操作不影响其子树的平衡因子。

首先删除节点，然后沿着该节点的父节点向根部更新平衡因子，考虑更新后的节点A

新的平衡因子，分为下面三种情况：

(1)、如果bf(A)=0，则高度减少了1，从而继续向上找非平衡点。

(2)、如果bf(A)=1或者-1，则之前必为0，从而不影响平衡性，结束。

(3)、如果bf(A)=2(原来为1)或者-2(原来为-1)，则A点非平衡。调整。

以bf(A)=-2为例，称A为L不平衡

如果A的右节点的平衡因子是0，则进行一次左旋转，如图：

L0

由于子树的高度和删除前一样，因此树已经平衡。

如果A的右节点的平衡因子是-1，则称为L-1不平衡，也要进行一次做旋转。

L-1

树的高度减少1，这使得上面的祖先节点可能不平衡，一次还要沿着路径在向上

寻找新的不平衡点，再次更新平衡因子，调整等。

如果A的右节点的平衡因子是1，则称为L1不平衡，要进行两次旋转。

L1

此时要根据BL的情况决定A和B的新的平衡因子。

由于树的高度减少了1，因此还要沿着路径继续向上寻找新的不平衡点。

至此树节点的删除操作完成。

总结：

归纳起来，AVL树的查找操作等同于一般二叉树，插入和删除操作除了一般的二叉树插入和删除

还要更新树的平衡因子，当平衡因子被打破时要通过旋转恢复。而且在调整平衡时尽量影响局部范围。

 

下面直接给出AVL树的实现代码，主要是基于前面的二叉查找树的基类实现方法的代码。

这里是直接修改原来的代码的，后面将会把它更新为继承的方式。

BinaryTreeNode.h 仅加入了一个balance平衡因子的数据成员和它的get,set方法

?
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
//此类为AVL查找树的树节点类 
//定义的关键子，值，父节点和儿子节点
#ifndef BINARY_TREE_NODE_H
#define BINARY_TREE_NODE_H
#include "objectclass.h"//通用类
classBinaryTreeNode
{
private:
    ObjectClass *theKey;//关键字
    ObjectClass *theValue;//值
    BinaryTreeNode *parent;//父亲节点
    BinaryTreeNode *left;//左儿子
    BinaryTreeNode *right;//右儿子
    //定义左右子树的宽度以便打印
    intleftWidth;
    intrightWidth;
    //定义当前节点应该输出的位子，从左起点到右的宽度
    intleftOutPutLen;
    //定义平衡因子
    intbalance;
public:
    BinaryTreeNode();
    BinaryTreeNode(ObjectClass *theKey,ObjectClass *theValue);
    ObjectClass *getKey();
    ObjectClass *getValue();
    BinaryTreeNode *getLeft();
    BinaryTreeNode *getRight();
    BinaryTreeNode *getParent();
    intgetLeftWidth();
    intgetRightWidth();
    intgetLeftOutPutLen();
    intgetBalance();
    voidsetKey(ObjectClass *theKey);
    voidsetValue(ObjectClass *theValue);  
    voidsetLeft(BinaryTreeNode *left);
    voidsetRight(BinaryTreeNode *Right);
    voidsetParent(BinaryTreeNode *parent);
    voidsetWidth(int,int);//设置子树宽度
    voidsetLeftOutPutLen(intlen);
    voidsetBalance(intbalance);
};
#endif

BinaryTreeNode.cpp 文件

?
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
#include "BinaryTreeNode.h"
BinaryTreeNode::BinaryTreeNode()
{
    theKey = NULL;
    theValue = NULL;
    parent = NULL;
    left = NULL;
    right = NULL;
    leftWidth=0;
    rightWidth=0;
    leftOutPutLen=0;
    balance=0;
}
BinaryTreeNode::BinaryTreeNode(ObjectClass *theKey,ObjectClass *theValue)
{
    this->theKey = theKey;
    this->theValue = theValue;
    parent = NULL;
    left = NULL;
    right = NULL;
    leftWidth=0;
    rightWidth=0;
    leftOutPutLen=0;
    balance=0;
}
int BinaryTreeNode::getLeftWidth()
{
    returnleftWidth;
}
int BinaryTreeNode::getRightWidth()
{
    returnrightWidth;
}
ObjectClass *BinaryTreeNode::getKey()
{
    returntheKey;
}
ObjectClass *BinaryTreeNode::getValue()
{
    returntheValue;
}
BinaryTreeNode *BinaryTreeNode::getLeft()
{
    returnleft;
}
BinaryTreeNode *BinaryTreeNode::getRight()
{
    returnright;
}
BinaryTreeNode *BinaryTreeNode::getParent()
{
    returnparent;
}
int BinaryTreeNode::getBalance()
{
    returnbalance;
}
voidBinaryTreeNode::setWidth(intleftWidth, intrightWidth)
{
    this->leftWidth=leftWidth;
    this->rightWidth=rightWidth;
}
voidBinaryTreeNode::setValue(ObjectClass *theValue)
{
    this->theValue = theValue;
}
voidBinaryTreeNode::setKey(ObjectClass *theKey)
{
    this->theKey = theKey;
}
voidBinaryTreeNode::setLeft(BinaryTreeNode *left)
{
    this->left = left;
}
voidBinaryTreeNode::setRight(BinaryTreeNode *right)
{
    this->right = right;
}
voidBinaryTreeNode::setParent(BinaryTreeNode *parent)
{
    this->parent=parent;
}
int BinaryTreeNode::getLeftOutPutLen()
{
    returnleftOutPutLen;
}
voidBinaryTreeNode::setLeftOutPutLen(intlen)
{
    this->leftOutPutLen = len;
}
voidBinaryTreeNode::setBalance(intbalance)
{
    this->balance = balance;
}

AVL.h 文件

?
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
//此类是AVL搜索树类的定义部分
#include "BinaryTreeNode.h"
classBSTree
{
private:
//根节点
BinaryTreeNode *root;
public:
BSTree();
public:
BinaryTreeNode *get(ObjectClass *theKey);//搜索
BinaryTreeNode *getRoot();//返回根节点
BinaryTreeNode *remove(ObjectClass *theKey);//删除
voidinsert(ObjectClass *theKey, ObjectClass *theValue);//插入
voidascend(BinaryTreeNode *);//遍历
int calWidth(BinaryTreeNode *);//计算各节点的长度
voidoutPut();//输出
BinaryTreeNode *tree_minimum(BinaryTreeNode *p);//最小节点
BinaryTreeNode *tree_maximum(BinaryTreeNode *p);//最大节点
BinaryTreeNode *tree_successor(BinaryTreeNode *p);//后继节点
voidleftRote(BinaryTreeNode *);
voidrightRote(BinaryTreeNode *);
voiddeleteNode(BinaryTreeNode *);
};

AVL.cpp文件

?
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
//此文件是AVL搜索树的实现部分
#include "AVL.h"
#include <iostream>
#include "queue.h"
usingnamespace std;
BSTree::BSTree()
{
    root = NULL;//根节点默认为NULL
}
//查找关键字为theKey的节点并返回指向节点的指针,找不到则返回空指针
BinaryTreeNode *BSTree::get(ObjectClass *theKey)
{
    BinaryTreeNode *p=root;
    while(p!=NULL)
    {
        //if(theKey < p->getKey())
        if(theKey->Compare(p->getKey()) == -1)
        p=p->getLeft();
        //if(theKey > p->getKey())
        elseif(theKey->Compare(p->getKey()) == 1)
        p=p->getRight();
        else//如果找到了相同的关键字则成功返回
        returnp;
    }
    returnNULL;
}
//插入一个节点，如果节点关键字已经存在则覆盖，否则插入到叶子节点处
voidBSTree::insert(ObjectClass *theKey,ObjectClass *theValue)
{
    BinaryTreeNode *firstBalanceNode=NULL;
    intPath=1;//记录firstBalanceNode开始往下的路径,以1开始为标记
    //00010010表示左左右左
    intk=0;//指示开始的位置,上例中k=4
    BinaryTreeNode *p=root;//search pointer
    BinaryTreeNode *parent=NULL;//parent of p;
    while(p!=NULL)
    {
        if(p->getBalance() != 0)
        {
           firstBalanceNode = p;
           Path=1;
           k=0;
        }
   
        parent=p;
        //if(theKey < p->getKey())
        if(theKey->Compare(p->getKey()) == -1)
        {
           k++;
           p=p->getLeft();
           Path=(Path<<1);//如果是向左走则Path后面加个0
        }
        //if(theKey > p->getKey())
        elseif(theKey->Compare(p->getKey()) == 1)
        {
           k++;
           p=p->getRight();
           Path=(Path<<1);
           Path=Path+1;//如果是向右走，则Path后面加个1
        }
        else
        {
           p->setValue(theValue);
           //如果找到相同的关键字则覆盖原有的
           return;
        }      
    }
    //等待插入的新节点
    BinaryTreeNode *newNode =new BinaryTreeNode(theKey,theValue);
    if(root == NULL)
       root = newNode;
    else
    {
        //当p为空的时候parent最多只有一个儿子节点
        //if(theKey < parent->getKey())
        if(theKey->Compare(parent->getKey()) == -1)
        {
           parent->setLeft(newNode);
           newNode->setParent(parent);
        }
        else
        {
           parent->setRight(newNode);
           newNode->setParent(parent);
        }
    }
    //第一种情况是没有找到这种非平衡点
    if(firstBalanceNode == NULL)
    {
        if(newNode == root)return;
        //自下而上一次更新平衡因子，直到根节点
        while(newNode!=NULL)
        {
           if(newNode->getParent() == NULL)break;
           if(newNode == newNode->getParent()->getLeft())
              newNode->getParent()->setBalance(newNode->getParent()->getBalance()+1);
           else
              newNode->getParent()->setBalance(newNode->getParent()->getBalance()-1);
           newNode = newNode -> getParent();
        }
        return;
    }
    //第二种情况非平衡点刚好因为新点的加入而变为平衡点，这不会影响上面节点的因子,
    //与第一种情形可以合并写，不过为了看起来清晰就分开做
    elseif((firstBalanceNode->getBalance() == -1 && ((Path>>(k-1))&1)==0)
        || (firstBalanceNode->getBalance() == 1 && ((Path>>(k-1))&1)==1))
    {
        //自下而上更新平衡因子，直到原来的非平衡点
        while(newNode!=firstBalanceNode)
        {
           if(newNode->getParent() == NULL)break;
           if(newNode == newNode->getParent()->getLeft())
              newNode->getParent()->setBalance(newNode->getParent()->getBalance()+1);
           else
              newNode->getParent()->setBalance(newNode->getParent()->getBalance()-1);
           newNode = newNode -> getParent();
        }
        return;
    }
    //第三种情况，AVL树被破坏掉了，要通过旋转恢复
     else
    {
        //首先依然是改变平衡因子
        while(newNode!=firstBalanceNode)
        {
           if(newNode->getParent() == NULL)break;
           if(newNode == newNode->getParent()->getLeft())
              newNode->getParent()->setBalance(newNode->getParent()->getBalance()+1);
           else
              newNode->getParent()->setBalance(newNode->getParent()->getBalance()-1);
           newNode = newNode -> getParent();
        }
        if(((Path>>(k-1))&1)==0 && ((Path>>(k-2))&1)==0)//LL
        {      
            firstBalanceNode->setBalance(0);
            firstBalanceNode->getLeft()->setBalance(0);
            rightRote(firstBalanceNode->getLeft());//右旋转
        }
        else if(((Path>>(k-1))&1)==1 && ((Path>>(k-2))&1)==1)//RR
        {
            firstBalanceNode->setBalance(0);
            firstBalanceNode->getRight()->setBalance(0);
            leftRote(firstBalanceNode->getRight());//左旋转
        }
        else if(((Path>>(k-1))&1)==0 && ((Path>>(k-2))&1)==1)//LR
        {
            int bal=firstBalanceNode->getLeft()->getRight()->getBalance();
            firstBalanceNode->getLeft()->getRight()->setBalance(0);
            if(bal == 0)
            {
               firstBalanceNode->setBalance(0);
               firstBalanceNode->getLeft()->setBalance(0);
            }
            else if(bal == 1)
            {
               firstBalanceNode->setBalance(-1);
               firstBalanceNode->getLeft()->setBalance(0);
            }
            else
            {
               firstBalanceNode->setBalance(0);
               firstBalanceNode->getLeft()->setBalance(1);
            }
            leftRote(firstBalanceNode->getLeft()->getRight());
            rightRote(firstBalanceNode->getLeft());                                                 
        }
        else if(((Path>>(k-1))&1)==1 && ((Path>>(k-2))&1)==0)//RL
        {
            int bal=firstBalanceNode->getRight()->getLeft()->getBalance();
            firstBalanceNode->getRight()->getLeft()->setBalance(0);
            if(bal == 0)
            {
               firstBalanceNode->setBalance(0);
               firstBalanceNode->getRight()->setBalance(0);
            }
            else if(bal == 1)
            {
               firstBalanceNode->setBalance(0);
               firstBalanceNode->getRight()->setBalance(-1);
            }
            else
            {
               firstBalanceNode->setBalance(1);//1
               firstBalanceNode->getRight()->setBalance(0);
            }
            rightRote(firstBalanceNode->getRight()->getLeft());
            leftRote(firstBalanceNode->getRight());                                                
        }
    }
    return;
}
//删除节点，如果这个节点含有少于两个儿子节点，则直接删除它，然后将它的儿子节点链接到它原来所在的位置
//如果这个节点含有两个儿子节点，则要先删除它的后继节点，然后将它的后继节点的值换给它
BinaryTreeNode *BSTree::remove(ObjectClass *theKey)
{
    //先查找到要删除的节点指针
    BinaryTreeNode *deletedNode=get(theKey);
    if(deletedNode==NULL)return NULL;
    //即将被删除的节点,注意这个节点最多只含有一个儿子节点
    BinaryTreeNode *todelete;
    //被删除节点的儿子节点
    BinaryTreeNode *nextNode;
    if(deletedNode->getLeft()==NULL || deletedNode->getRight()==NULL)
        //当要删除的节点只含有最多一个儿子节点时则即将被删除节点就是要删除的节点
        todelete = deletedNode;
    else
        todelete = tree_successor(deletedNode);//否则的话删除它的后继节点
    //////////////////////////////////////////////////////////////////////////////
    //这一步是为了更新平衡因子调整树的结构而设计的
    BinaryTreeNode *pend=todelete;
    while(pend!=NULL)
    {
        if(pend->getParent()==NULL)break;//根节点
        if(pend == pend->getParent()->getLeft())
            pend->getParent()->setBalance(pend->getParent()->getBalance()-1);
        else
            pend->getParent()->setBalance(pend->getParent()->getBalance()+1);
        pend=pend->getParent();
        if(pend->getBalance() != 0)break;
    }
    //////////////////////////////////////////////////////////////////////////////
    //获取唯一的儿子节点，准备当前即将删除节点的删除工作
    if(todelete->getLeft()!=NULL)
        nextNode=todelete->getLeft();
    else
        nextNode=todelete->getRight();
    //开始删除节点
    if(nextNode!=NULL)
        nextNode->setParent(todelete->getParent());
    if(todelete->getParent()==NULL)
        root=nextNode;
    elseif(todelete->getParent()->getLeft()==todelete)
        todelete->getParent()->setLeft(nextNode);
    else
        todelete->getParent()->setRight(nextNode);
    //节点成功删除，删完后在考虑将原来节点的后续节点值的替换
    if(todelete!=deletedNode)
    {
        deletedNode->setKey(todelete->getKey());
        deletedNode->setValue(todelete->getValue());
    }
    //删除节点
    deletetodelete;
    //更新平衡因子
    deleteNode(pend);
    //返回不平衡点
    returnpend;
}
//删除节点函数的辅助函数，用于更新平衡因子，调整树的结构，使之仍为AVL树
voidBSTree::deleteNode(BinaryTreeNode *deletedNode)
{
    if(deletedNode==NULL)return;
    if(deletedNode->getBalance()==1 || deletedNode->getBalance()==-1//如果非平衡点删除之前的平衡因子是0，则无需调整
        || deletedNode->getBalance()==0)return;//如果非平衡点是根节点且平衡因子是0则无需调整
    if(deletedNode->getBalance()==2)//R
    {
        if(deletedNode->getLeft()->getBalance()==0)//R0
        {
            deletedNode->setBalance(1);
            deletedNode->getLeft()->setBalance(-1);
            rightRote(deletedNode->getLeft());
            return;// 经过R0旋转后，AVL树已经达到平衡所以直接返回即可
        }
        else if(deletedNode->getLeft()->getBalance()==1)//R-1
        {
            deletedNode->setBalance(0);
            deletedNode->getLeft()->setBalance(0);
            rightRote(deletedNode->getLeft());
        }
        else//R1
        {
            if(deletedNode->getLeft()->getRight()->getBalance()==0)
            {
               deletedNode->setBalance(0);
               deletedNode->getLeft()->setBalance(0);
            }
            else if(deletedNode->getLeft()->getRight()->getBalance()==1)
            {
               deletedNode->setBalance(-1);
               deletedNode->getLeft()->setBalance(0);
            }
            else
            {
               deletedNode->setBalance(0);
               deletedNode->getLeft()->setBalance(1);
            }
            deletedNode->getLeft()->getRight()->setBalance(0);
            leftRote(deletedNode->getLeft()->getRight());
            rightRote(deletedNode->getLeft());
        }
        }
    elseif(deletedNode->getBalance()==-2)//L
    {
        if(deletedNode->getRight()->getBalance()==0)//L0
        {
            deletedNode->setBalance(-1);
            deletedNode->getRight()->setBalance(1);
            leftRote(deletedNode->getRight());
            return;
        }
        else if(deletedNode->getRight()->getBalance()==-1)//L-1
        {
            deletedNode->setBalance(0);
            deletedNode->getRight()->setBalance(0);
            leftRote(deletedNode->getRight());
        }
        else//L1
        {
            if(deletedNode->getRight()->getLeft()->getBalance()==0)
            {
               deletedNode->setBalance(0);
               deletedNode->getRight()->setBalance(0);
            }
            else if(deletedNode->getRight()->getLeft()->getBalance()==-1)
            {
               deletedNode->setBalance(1);
               deletedNode->getRight()->setBalance(0);
            }
            else
            {
               deletedNode->setBalance(0);
               deletedNode->getRight()->setBalance(-1);
            }
            deletedNode->getRight()->getLeft()->setBalance(0);
            rightRote(deletedNode->getRight()->getLeft());
            leftRote(deletedNode->getRight());
        }
    }
    //如果不是0旋转则继续向根部寻找非平衡点，
    deletedNode=deletedNode->getParent();//首先从当前的被调整后的子树根节点开始出发
    while(deletedNode!=NULL)
    {
        if(deletedNode->getParent()==NULL)break;//根节点
        if(deletedNode == deletedNode->getParent()->getLeft())
            deletedNode->getParent()->setBalance(deletedNode->getParent()->getBalance()-1);
        else
            deletedNode->getParent()->setBalance(deletedNode->getParent()->getBalance()+1);
        deletedNode=deletedNode->getParent();
        if(deletedNode->getBalance() != 0)break;//找到非平衡点就退出，否则会一直找打根部
    }
    deleteNode(deletedNode);//继续下一次更新
}
//计算左右的宽度，使用递归算法
intBSTree::calWidth(BinaryTreeNode *p)
{
    if(p!=NULL)
    {
        int leftWidth=0;//左宽度
        int rightWidth=0;//右宽度
        if(p->getLeft()!=NULL)
            leftWidth=calWidth(p->getLeft())+p->getKey()->getLength();
            //左宽度是左子树的总宽度加上本节点的长度
        if(p->getRight()!=NULL)
            rightWidth=calWidth(p->getRight())+p->getKey()->getLength();
            //右宽度是右子树的总宽度加上本节点的长度
        p->setWidth(leftWidth,rightWidth);//设置左右宽度
        //返回本节点为根的子树总宽度
        return leftWidth+rightWidth;
    }
    return0;
}
//按照层次遍历子树并且打印出来
voidBSTree::ascend(BinaryTreeNode *p)
{  
    if(p==NULL)return;
    calWidth(p);//计算左右子树的宽度
    p->setLeftOutPutLen(p->getLeftWidth());//设置最顶层的左边预留宽度
    //下面要用队列实现树的层次遍历
    Queue<BinaryTreeNode *> Q;
    Q.EnQueue(p);
    intnumber=0;//存储下一层的元素个数
    intnumberLeave=1;//这一层还剩下多少元素
    BinaryTreeNode *dep;//保存当前从队列弹出的节点指针
    intpreLeftWidth=0;
    //存储前一个节点的左宽度，以便后面一个节点的打印
    //如果当前节点在最开始，则前一节点左宽度为0
    boolfirstIn=true;
    while(!Q.isEmpty())//打印所有节点
    {
        dep=Q.DeQueue();
        numberLeave--;
        if(dep!=NULL)
        {
            if(dep->getLeft()!=NULL)
            {
               Q.EnQueue(dep->getLeft());//左节点加入队列
               number++;//下层节点加一
            }
           
            if(dep->getRight()!=NULL)
            {
               Q.EnQueue(dep->getRight());//右节点加入队列
               number++;//下层节点加一
            }
            int leftOutPutLen=dep->getLeftWidth();
            //如果是第一次进入就左边预留宽度就是当前节点自己的宽度
           
            if(!firstIn)
            {
               if(dep==dep->getParent()->getLeft())
                   leftOutPutLen = dep->getParent()->getLeftOutPutLen()-dep->getRightWidth()-dep->getParent()->getKey()->getLength();
               //如果当前节点是左儿子，则它的左预留宽度是父节点的预留宽度减去当前节点右宽度
               else
                   leftOutPutLen = dep->getParent()->getLeftOutPutLen()+dep->getLeftWidth()+dep->getParent()->getKey()->getLength();
               //如果当前节点是右儿子，则它的左预留宽度是父节点的预留宽度加上当前节点的左宽度
               dep->setLeftOutPutLen(leftOutPutLen);//设置预留宽度
            }
            //根据当前节点左预留宽度和上一兄弟节点的结束宽度打印预留空格
            for(inti=0;i<leftOutPutLen-preLeftWidth;i++)
               cout<<" ";
            dep->getKey()->OutPut();//打印当前节点
            preLeftWidth=leftOutPutLen+dep->getKey()->getLength();
            //计算当前节点的结束宽度，以便下一兄弟节点的打印
            //如果当前节点在没有兄弟节点了就换行
            if(numberLeave == 0)
            {
               cout<<endl;
               preLeftWidth=0;
               numberLeave = number;
               number = 0;
            }
        }
        firstIn=false;
    }
}
//输出，这里是默认从根节点输出，如果直接调用ascend则可以输出任何子树
voidBSTree::outPut()
{
    BinaryTreeNode *temp=root;
    ascend(temp);
}
BinaryTreeNode *BSTree::tree_minimum(BinaryTreeNode *p)
{
    if(p==NULL)
    returnNULL;
    BinaryTreeNode *pp=p;
    while(pp->getLeft()!=NULL)
       pp=pp->getLeft();
    returnpp;
}
BinaryTreeNode *BSTree::tree_maximum(BinaryTreeNode *p)
{
    if(p==NULL)
    returnNULL;
    BinaryTreeNode *pp=p;
    while(pp->getRight()!=NULL)
       pp=pp->getRight();
    returnpp;
}
//返回已知节点的后续节点
//如果这个节点有右子数，则返回右子树的最小节点
//否则向父节点寻找，找到第一个向右转的父节点为止
BinaryTreeNode *BSTree::tree_successor(BinaryTreeNode *p)
{
    if(p==NULL)return NULL;
       BinaryTreeNode *pp=p;
    if(pp->getRight()!=NULL)
       return tree_minimum(pp->getRight());
    BinaryTreeNode *y=p->getParent();
    while(y!=NULL && pp==y->getRight())
    {
       pp=y;
       y=y->getParent();
    }
    returny;
}
BinaryTreeNode *BSTree::getRoot()
{
    returnroot;
}
//左旋转，这里需要特别注意根节点的变化
voidBSTree::leftRote(BinaryTreeNode *p)
{
    BinaryTreeNode *parent=p->getParent();
    if(parent==root)
    {
        root=p;
        p->setParent(NULL);
    }
    else
    {
        p->setParent(parent->getParent());
        if(parent->getParent()->getLeft()==parent)
            parent->getParent()->setLeft(p);
        else
            parent->getParent()->setRight(p);
    }
    parent->setRight(p->getLeft());
    if(p->getLeft()!=NULL)
        p->getLeft()->setParent(parent);
    p->setLeft(parent);
    parent->setParent(p);
}
//右旋转
voidBSTree::rightRote(BinaryTreeNode *p)
{  
    BinaryTreeNode *parent=p->getParent();
    if(parent==root)
    {
    root=p;
    p->setParent(NULL);
    }
    else
    {
        p->setParent(parent->getParent());
        if(parent->getParent()->getLeft()==parent)
        parent->getParent()->setLeft(p);
        else
        parent->getParent()->setRight(p);
    }
    parent->setLeft(p->getRight());
    if(p->getRight()!=NULL)
        p->getRight()->setParent(parent);
    p->setRight(parent);
    parent->setParent(p);
}
 

下面用一个例子来测试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
#include <iostream>
#include "AVL.h"
#include "IntClass.h"
#include <time.h>
#include <stdlib.h>
#include "StringClass.h"
usingnamespace std;
int main()
{
BSTree bstree;
BSTree bstreeStr;
int max=50;
srand(unsigned(time(0)));
for(intj=0;j<50;j++)
{
    intnumber=rand()%7;
    for(inti=number;i>0;i--)
    {
        inta=rand()%max;
        bstree.insert(newIntClass(a),newStringClass("Tao",3));
    }
    for(inti=number;i>0;i--)
    {
        inta=rand()%max;
        bstree.remove(newIntClass(a));
    }
}
bstree.outPut();
cout<<"------------------------------------------------"<<endl;
bstreeStr.insert(newStringClass("Jim",3),newStringClass("Hello, I'm a student",20));
bstreeStr.insert(newStringClass("Lucy",4),newStringClass("Hello, I'm a teacher",20));
bstreeStr.insert(newStringClass("Brown",5),newStringClass("Hello, I'm a doctor",19));
bstreeStr.insert(newStringClass("Lily",4),newStringClass("Hello, I'm a actor",18));
bstreeStr.insert(newStringClass("Tao",3),newStringClass("Hello, I'm a student",20));
bstreeStr.insert(newStringClass("Peter",5),newStringClass("Hello, I'm a teacher",20));
bstreeStr.insert(newStringClass("John",4),newStringClass("Hello, I'm a doctor",19));
bstreeStr.insert(newStringClass("Tony",4),newStringClass("Hello, I'm a actor",18));
bstreeStr.insert(newStringClass("Linda",5),newStringClass("Hello, I'm a student",20));
bstreeStr.insert(newStringClass("Jurcy",5),newStringClass("Hello, I'm a teacher",20));
bstreeStr.insert(newStringClass("Chern",5),newStringClass("Hello, I'm a doctor",19));
bstreeStr.outPut();
cout<<"-------------------------------------------------"<<endl;
return0;
}

结果如下图，可见AVL树的平衡性很好，几乎达到完全二叉树的效果

AVL

 

 

 

银行卡被网络盗刷 银行卡被网络盗刷
原创粉丝点击
热门IT博客
奥卡姆剃刀通信知乎奥卡姆剃刀通信知乎
c语言指针看什么书c语言指针看什么书
冰镇西瓜 知乎
mysql 服务器已离线
兼职淘宝扫二维码骗局
淘宝买摩托车哪家的好
劳氏船舶数据
报了java培训班学不会
自动扒带软件
windows live香港官网
linux 资源暂时不可用
用什么软件写python
修正主义 知乎
linux java native
网络语言暴力的影响
php记录点击次数
linux 查看文件夹详情
网络运营公司
淘宝组装机能买吗
七剑战歌之风月 知乎
热门问题 老师的惩罚 人脸识别 我在镇武司摸鱼那些年 重生之率土为王 我在大康的咸鱼生活 盘龙之生命进化 天生仙种 凡人之先天五行 春回大明朝 姑娘不必设防，我是瞎子 低分子 一个氢分子 分子理论 分子相同 分子的定义 分子的性质 信息分子 积极分子思想汇报 相对分子质量 积极分子汇报 低分子肝素钙 大学生积极分子思想汇报 dna分子的结构 积极分子考察表 硫酸铜相对分子质量 碳酸钠相对分子质量 积极分子思想汇报2019 低分子肝素钠 积极分子考察意见 积极分子思想汇报格式 硫酸相对分子质量 n相对分子质量 氯化钠相对分子质量 碳酸钙相对分子质量 积极分子思想汇报2018 分子原子本质区别 2019积极分子思想汇报 积极分子培养考察情况 相对分子质量单位 硫酸钠相对分子质量 水分子由什么构成 小分子胶原蛋白肽价格 气相分子吸收光谱仪 超高分子聚乙烯板厂家 分子间相互作用 分子生物学实验室 胶原蛋白小分子肽 超高分子聚乙烯板价格 注射小分子玻尿酸 分子 原子 离子 超高分子聚乙烯管

程序博客网，程序员的互联网技术博客家园。csdn论坛精品 msdn技术资料都在这里