前几年写的平衡二叉树程序（C++）

C++的平衡二叉树程序(暂时没有加入调整算法)

  1/**//////////////////////////////////////////////////////////////////////////////
  2//程序名称：AVL平衡二叉树构建程序                 //
  3//文件名称：avl.cpp                                            //
  4//作    者：CityWildWolf                                    //
  5//建立时间：2007-5-29                                    //
  6//修改时间：                                                       //
  7//程序说明：                                                       //
  8//                                                    //
  9//                                                        //
 10/**/////////////////////////////////////////////////////////////////////////////
 11
 12
 13#include <iostream>
 14
 15#define NODELEFT -1                //指示插入的位置为工作指针的左孩子
 16#define NODERIGHT 1                //指示插入的位置为工作指针的右孩子
 17
 18using namespace std;
 19
 20
 21
 22//二叉树数据结构
 23struct   AvlBitree
 24{
 25  
 26    int   value;                   //结点值
 27    int blanceRank;                //结点平衡因子
 28    struct   AvlBitree   *right;   //左孩子
 29    struct   AvlBitree   *left;    //右孩子
 30};
 31
 32//函数原型
 33int PrintMenu();
 34AvlBitree* BuildTree(AvlBitree *);
 35AvlBitree* CreatNode(int );
 36void InsertNode(AvlBitree *,AvlBitree *);
 37void Traveser(AvlBitree *r);
 38
 39
 40//主函数，程序入口
 41int main()
 42{
 43    //定义一个根结点
 44    AvlBitree *root = NULL;
 45    //在屏幕打印菜单，让用户选择建立、遍历、查找功能
 46    int userChoseMenu;
 47    while (userChoseMenu = PrintMenu())
 48    {
 49        //
 50        switch (userChoseMenu)
 51        {
 52            case 1 :
 53                cout << "你选择了1，开始建AVL！" << endl;
 54                root = BuildTree(root);
 55                break;
 56            case 2 :
 57                cout << "你选择了2，给AVL加结点！" << endl;
 58                break;
 59            case 3 :
 60                cout << "你选择了3，遍历输出AVL。" << endl;
 61                Traveser(root);
 62                //cout << root->value;
 63                break;
 64            case 4 :
 65                cout << "你选择了4！" << endl;
 66                break;
 67            default :
 68                cout << "你选择了其他！" << endl;
 69                break;
 70        }
 71    }
 72  
 73    //程序结束，退出程序
 74    //
 75    return 0;
 76}
 77
 78//在屏幕打印菜单函数
 79int PrintMenu()
 80{
 81    int userChose;
 82    cout << "-----------------------" << endl;
 83    cout << "1. Creat a New AVL Tree" << endl;
 84    cout << "2. Add New Node To AVL " << endl;
 85    cout << "3. Travesers the AVL   " << endl;
 86    cout << "4. Find The Node In AVL" << endl;
 87    cout << "-----------------------" << endl;
 88    cout << "Enter Your Chose:";
 89    cin >> userChose;
 90    return userChose;
 91}
 92
 93
 94//新建二叉树函数
 95AvlBitree* BuildTree(AvlBitree *r)
 96{
 97    //如果传入根结点不为空，则树已构建过，退出函数
 98    if (r != NULL)
 99    {
100        cout << "AVL Tree has been built!";
101        return NULL;
102    }
103  
104    //根结点为空则开始构建
105    //AvlBitree *p = *r;
106    cout << "请输入结点值，输入零则结束" << endl;
107    int nodeValue;
108    cin >> nodeValue;
109    while (nodeValue)
110    {
111        //建立新结点
112        AvlBitree *node = CreatNode(nodeValue);
113        //如果根为空，则将此结点设置为根结点
114        if (r == NULL)
115        {
116            r = node;
117        }
118        //否则将此结点作为非根结点插入
119        else
120        {
121            InsertNode(r,node);
122        }
123        //输入新值
124        cin >> nodeValue;
125    }
126    return r;
127  
128}
129
130//插入结点函数
131void InsertNode(AvlBitree *r,AvlBitree *p)
132{
133    //插入当前结点
134    //r为根结点，p为新插入结点
135    //定义一个q作为当前工作指针，定义一个指针t指向q的父结点
136    AvlBitree *t,*q;
137    //t,q都指向根结点
138    t = q = r;
139    //定一个哨兵flag，当flag=1时，在右孩子处加入结点，当flag=-1时在左孩子处加入
140    int flag;
141    //开始循环查找p应该插入的位置，当工作指针q指向空时，此位置为插入位置
142    while (q != NULL)
143    {
144        if(p->value > q->value)
145        {
146            //右移
147            t = q;
148            q = q->right;
149            //cout << "Right once" << endl;
150            flag = NODERIGHT;
151        }
152        else if (p->value < q->value)
153        {
154            //左移
155            t = q;
156            q = q->left;
157            //cout << "Left once" << endl;
158            flag = NODELEFT;
159        }
160        else
161        {
162            return;
163        }
164    }
165    //插入该结点
166    //如果该插入的位置为t的左孩子
167    if (flag == NODERIGHT)
168    {
169        t->right = p;
170    }
171    //如果该插入的位置为t的右孩子
172    else if(flag == NODELEFT)
173    {
174        t->left = p;
175    }  
176}
177
178
179
180//递归中跟遍历二叉树
181void Traveser(AvlBitree *r)
182{   
183     if(r->left) Traveser(r->left);
184     if(r) cout << r->value << endl;
185     if(r->right) Traveser(r->right);   
186}
187
188//建立新结点
189AvlBitree* CreatNode(int nodeValue)
190{
191    AvlBitree *node;
192    node = (AvlBitree *)malloc(sizeof(AvlBitree));
193    node->value = nodeValue;
194    node->blanceRank = 0;
195    node->left = NULL;
196    node->right = NULL;
197    return node;
198}