C++实现AVL树的完整代码
更新时间：2021年06月02日 10:28:41 作者：QS小其
AVL树是高度平衡的而二叉树。它的特点是：AVL树中任何节点的两个子树的高度最大差别为1。 今天通过本文给大家分享C++实现AVL树的完整代码，感兴趣的朋友一起看看吧
AVL树的介绍
AVL树是一种自平衡的二叉搜索树，它通过单旋转（single rotate）和双旋转(double rotate)的方式实现了根节点的左子树与右子树的高度差不超过1,。这有效的降低了二叉搜索树的时间复杂度，为O（log n）。那么，下面小编将详细介绍C++实现AVL树的代码。最后一步提供可靠的代码实现
这里先粘贴代码
给大家的忠告，一定要及时去实现，不然之后再实现要花更多的时间
/*
*平衡二叉树应该有些功能
*插入 删除 查找
*前序遍历 中序遍历 后序遍历 层次遍历
*统计结点数目
*/
//代码已经调好，写了很久才写出来
#ifndef _AVLTREE_
#define _AVLTREE_
#include<iostream>
#include<vector>
#include<queue>
#include<map>
using namespace std;
#define MAXFACTOR = 2;
template<class Key , class E>
class AVLNode
{
private:
Key key;
E value;
AVLNode<Key,E>* left;
AVLNode<Key,E>* right;
AVLNode<Key,E>* parent;
public:
AVLNode():left(nullptr),right(nullptr),parent(nullptr){}
AVLNode(Key _key,E _value , AVLNode<Key,E>* _parent = nullptr,AVLNode<Key,E>*_left = nullptr, AVLNode<Key,E>*_right = nullptr):
key(_key),value(_value),left(_left),right(_right),parent(_parent){}
bool isLeaf(){return left==nullptr && right == nullptr ;}
//元素设置
Key getKey() const { return key;}
void setKey(Key set) {key = set;}
E getValue() const { return value;}
void setValue(E set) {value = set;}
AVLNode<Key,E>* getLeft() { return left; }
void setLeft (AVLNode< Key,E >* set){ left = set;}
AVLNode<Key,E>* getRight() { return right;}
void setRight (AVLNode<Key,E>* set) {right = set ;}
AVLNode<Key,E>* getParent() {return parent;}
void setParent(AVLNode<Key,E>* set) { parent = set;}
};
template<class Key , class E>
class AVLTree
{
private:
AVLNode<Key,E>* root;
void clear(AVLNode<Key,E>* &r)
{
if(r==nullptr)return;
if(r->getLeft())clear(r->getLeft());
if(r->getRight())clear(r->getRight);
delete r;
}
void Init()
{
root = new AVLNode<Key,E>();
root=nullptr;
}
void preOrder(AVLNode<Key,E>* r,void(*visit) (AVLNode<Key,E> * node))
{
if(r==nullptr)return;
(*visit) (r);
preOrder(r->getLeft() , visit);
preOrder(r->getRight(), visit);
}
void inOrder(AVLNode<Key,E>* r , void(*visit)(AVLNode<Key,E>* node) )
{
if(r==nullptr)return;
inOrder(r->getLeft() , visit);
(*visit)(r);
inOrder(r->getRight(),visit);
}
void postOrder(AVLNode<Key,E>*r , void (*visit) (AVLNode<Key,E>* node))
{
if(r==nullptr)return;
postOrder(r->getLeft(),visit);
postOrder(r->getRight(),visit);
(*visit)(r);
}
void levelOrder(AVLNode<Key,E>*r , void (*visit) (AVLNode<Key,E>* node))
{
queue< AVLNode<Key,E>* > q;
if(r==nullptr)return;
q.push(r);
while( ! q.empty() )
{
AVLNode<Key,E>* tmp = q.front();
q.pop();
(*visit)(tmp);
if(tmp->getLeft() ) q.push(tmp->getLeft() );
if(tmp->getRight()) q.push(tmp->getRight());
}
}
AVLNode<Key,E>* find(AVLNode<Key,E>* r,Key k)
{
if(r==nullptr)return nullptr;
if(k == r->getKey() ) return r;
else if( k < r->getKey())
{
find(r->getLeft(),k);
}
else {
find(r->getRight(),k);
}
}
//Find the smallest element in the avl tree
AVLNode<Key,E>* getMin(AVLNode<Key,E>* r)
{
if(r->getLeft() == nullptr) return r;
else{
getMin(r->getLeft());
}
}
//Remove the smallest element
AVLNode<Key,E>* deleteMin(AVLNode<Key,E>* rt)
{
if(rt->getLeft() == nullptr) return rt->getRight();
else{
rt->setLeft(deleteMin(rt->getLeft()));
return rt;
}
}
AVLNode<Key,E>* leftRotate(AVLNode<Key,E>* node)
{
if( node == nullptr) return nullptr;
AVLNode<Key,E>* newHead = node->getRight();
node->setRight( newHead -> getLeft() );
newHead -> setLeft( node );
return newHead;
}
AVLNode<Key,E>* rightRotate(AVLNode<Key,E>* node)
{
if(node == nullptr)return nullptr;
AVLNode<Key,E>* newHead = node->getLeft();
node->setLeft( newHead->getRight() );
newHead ->setRight(node);
return newHead;
}
int getHeight(AVLNode<Key,E>*node)
{
if(node == nullptr)return 0;
if(node->isLeaf())return 1;
else return ( getHeight( node->getLeft() ) > getHeight( node->getRight() ) ?
getHeight( node->getLeft() ) : getHeight( node->getRight() ) ) + 1;
}
int getBalanceFactor(AVLNode<Key,E>* node)
{
return getHeight(node->getLeft()) - getHeight(node->getRight() );
}
AVLNode<Key,E>* balance(AVLNode<Key,E>* node)
{
if(!node) return nullptr;
else if ( getBalanceFactor( node ) == 2)
{
if(getBalanceFactor( node ->getLeft() ) == 1)
{
node = rightRotate(node);
}
else
{
node->setLeft(leftRotate( node->getLeft() ) );
node = rightRotate(node);
}
}
else if(getBalanceFactor( node ) == -2)
{
if(getBalanceFactor( node->getRight()) == -1)
{
node = leftRotate(node);
}
else
{
node->setRight( rightRotate( node->getRight() ) );
node = leftRotate(node);
}
}
return node;
}
AVLNode<Key,E>* insert( AVLNode<Key,E>* root ,const pair<Key,E>& it)
{
if(root == nullptr)
{
return new AVLNode<Key,E>(it.first , it.second,NULL,NULL,NULL);
}
else if (it.first < root->getKey() )
{
root ->setLeft( insert(root->getLeft() , it) ) ;
}
else{
root ->setRight( insert(root->getRight() , it) );
}
root = balance(root);
return root;
}
AVLNode<Key,E>* remove(AVLNode<Key,E>* node , const Key k)
{
if(node == nullptr) return nullptr;
if(node->getKey() > k)
{
node->setLeft( remove(node->getLeft() , k) );
node = balance(node);
}
else if(node->getKey() < k)
{
node->setRight( remove(node->getRight(), k) );
node = balance(node);
}
else if(node->getKey() == k)
{
if(! node->isLeaf() )
{
AVLNode<Key,E>* tmp = getMin(node->getRight() );
node->setKey( tmp->getKey() );
node->setValue( tmp->getValue() );
node->setRight( deleteMin(node->getRight() ) );
delete tmp;
}
else {
AVLNode<Key,E>* tmp = node;
node = (node->getLeft() != nullptr) ? node->getLeft() : node->getRight() ;
delete tmp;
}
}
return node;
}
public:
~AVLTree(){clear(root);}
AVLTree(){/*Init();*/ root = nullptr; }
//四种遍历方式
void preOrder( void (*visit)(AVLNode<Key,E>* r))
{
preOrder(root,visit);
}
void inOrder(void (*visit)(AVLNode<Key,E>* r))
{
inOrder(root,visit);
}
void postOrder(void (*visit)(AVLNode<Key,E>* r))
{
postOrder(root,visit);
}
void levelOrder( void(*visit)(AVLNode<Key,E>*r) )
{
levelOrder(root,visit);
}
//插入
void insert(const pair<Key,E> &it)
{
root = insert(root,it);
}
//删除
void remove(const Key& k)
{
remove(root,k);
}
bool find(const Key&k)
{
return find(root,k);
}
};
#endif
//AVLtest.cpp
#include"NewAvl.h"
#include<iostream>
using namespace std;
template<typename Key,typename E>
void traverse(AVLNode<Key,E>* root)
{
cout<<root->getKey()<<" "<<root->getValue()<<" ";
cout<<endl;
}
int main()
{
AVLTree<int,int>* tree = new AVLTree<int ,int>;
for(int i = 0 ; i < 5 ; i ++)
{
tree->insert(make_pair(i,i));
}
tree->remove(1);
cout<<"PreOrder: "<<endl;
tree->preOrder(traverse);
cout<<endl;
cout<<"LevelOrder: "<<endl;
tree->levelOrder(traverse);
cout<<endl;
cout<<"InOrder: "<<endl;
tree->inOrder(traverse);
cout<<endl;
cout<<"PostOrder: "<<endl;
tree->postOrder(traverse);
cout<<endl;
cout<<tree->find(2)<<endl;
tree->insert(make_pair(9,9));
tree->levelOrder(traverse);
}
运行结果
以上就是C++实现AVL树的完整代码的详细内容，更多关于C++ AVL树的资料请关注脚本之家其它相关文章！
相关文章
这篇文章主要介绍了C++实现LeetCode(228.总结区间),本篇文章通过简要的案例,讲解了该项技术的了解与使用,以下就是详细内容,需要的朋友可以参考下2021-07-07
内存管理是C++最令人切齿痛恨的问题，也是C++最有争议的问题，C++高手从中获得了更好的性能，更大的自由，今天给大家分享C/C++内存管理的实例代码，需要的朋友参考下吧2021-06-06
大家好，本篇文章主要讲的是C语言实现第一次防死版扫雷游戏，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下2022-01-01
这篇文章主要介绍了C语言中如何实现单链表删除指定结点，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-07-07
今天小编就为大家分享一篇关于C++函数与指针的文章，小编觉得内容挺不错的，现在分享给大家，具有很好的参考价值，需要的朋友一起跟随小编来看看吧2021-10-10
JetBrains CLion永久激活超详细教程(最新激活方法)
JetBrains Clion 是一款专为 C/C++ 开发所设计的跨平台 IDE，本文适用 JetBrains CLion v2019.3/3.1/3.2/3.3 永久激活，附破解补丁和激活码，可以永久激活 Windows、MAC、Linux 下的 CLion，下面给大家分享JetBrains CLion永久激活超详细教程，感兴趣的朋友一起看看吧2023-01-01
本篇文章是对C++读写Excel的实现方法进行了详细的分析介绍，需要的朋友参考下2013-05-05
这篇文章主要介绍了C语言结构体指针案例解析,本文通过例子来解释说明了C语言的结构体概念和如何用指针去操作结构体,文章标明了详细的代码,需要的朋友可以参考下2021-07-07
这篇文章主要介绍了C语言枚举的使用以及使用，阅读下面内容我们将掌握枚举的相关概念、掌握枚举的几种用法、掌握枚举在实际产品中的用法,需要的朋友可以参考一下2022-03-03
互斥量是一种同步原语,用于保护多个线程同时访问共享数据,互斥量提供独占的、非递归的所有权语义,本文将和大家一起深入探究C/C++中互斥量(锁)的实现原理,感兴趣的小伙伴跟着小编一起来看看吧2024-06-06
最新评论