关于C++类库KYLib: 用C语言实现平衡二叉树(AVL tree)的源码

平衡二叉树(AVL tree)调整算法请参见我的博文: <关于平衡二叉树(AVL tree)旋转后平衡标志调整的计算公式>

若要在 C++ 中使用则只要将 KYAVLTreeC.c 改为 KYAVLTreeC.cpp 即可。

用C语言实现平衡二叉树(AVL tree)头文件如下:

// ======================================= // Unit : AVL tree (KYAVLTreeC.h) // Version: 2.1.0.0 (build 2011.03.04) // Author : Kyee Ye // Email : kyee_ye(at)126.com // Copyright (C) Kyee workroom // ======================================= #ifndef _KYAVLTreeC_H_ #define _KYAVLTreeC_H_ // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ /* 类型定义 */ // 布尔类型 {0: False, 1: True} #ifndef _KYBOOL_DEFINED_ typedef char Bool; #define _KYBOOL_DEFINED_ #endif // AVL 树结点类型 typedef struct { void* Item; // 项数据 void* Data; // 自定义数据 } TKYAVLNode, *PKYAVLNode; // OnCompare 结点比较事件 // 若返回值 result == 0 则表示 ANode1 等于 ANode2 // 若返回值 result > 0 则表示 ANode1 大于 ANode2 // 若返回值 result < 0 则表示 ANode1 小于 ANode2 typedef long (*TKYAVL_OnCompare)(const TKYAVLNode* ANode1, const TKYAVLNode* ANode2); // OnDeletion 结点删除事件 typedef void (*TKYAVL_OnDeletion)(void* ATree, TKYAVLNode* ANode); // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ /* 公用函数, 注: 为了提高效率不检查 void* ATree 是否合法, 此参数需要外部来维护 */ // 创建/释放 AVL 树 void* KYAVLTree_Create(TKYAVL_OnCompare OnCompare, TKYAVL_OnDeletion OnDeletion); void KYAVLTree_Free(void* ATree); // 读取 AVL 树属性 void* KYAVLTree_Data(void* ATree); // default: NULL long KYAVLTree_Count(void* ATree); // default: 0 long KYAVLTree_MaxCount(void* ATree); // default: 0 Bool KYAVLTree_CanDuplicate(void* ATree); // default: False // 设置 AVL 树属性 void KYAVLTree_SetData(void* ATree, void* AData); void KYAVLTree_SetMaxCount(void* ATree, long AMaxCount); void KYAVLTree_SetCanDuplicate(void* ATree, Bool ACanDuplicate); // 清除 AVL 树的所有结点, 激发 OnDeletion 事件 void KYAVLTree_Clear(void* ATree); // 添加 AVL 结点, 同时: {Node->Item == AItem, Node->Data == Data} TKYAVLNode* KYAVLTree_Add(void* ATree, void* AItem, void* AData); // 删除值为 AItem 和 Data 的 Compare(...) == 0 第一个 AVL 结点, 激发 OnDeletion 事件 Bool KYAVLTree_Delete(void* ATree, const void* AItem, const void* AData); // 移除指定 AVL 结点, 激发 OnDeletion 事件 Bool KYAVLTree_Remove(void* ATree, TKYAVLNode* ANode); // 搜索值为 AItem 和 Data 的 Compare(...) == 0 第一个 AVL 结点 TKYAVLNode* KYAVLTree_Find(void* ATree, const void* AItem, const void* AData); Bool KYAVLTree_Search(void* ATree, void** ARetItem, void** ARetData, const void* AItem, const void* AData); // 查找最近一个 AVL 结点 // 若 ANearest == NULL 则表示项值大于树中的最后一个结点值; // 若返回值为 true, 则表示找到项值的结点, 否则项值在 ANearest 结点之前 Bool KYAVLTree_FindNearest(void* ATree, const void* AItem, const void* AData, PKYAVLNode* ANearest); // 判断 AVL 结点是否存在 Bool KYAVLTree_Existed(void* ATree, TKYAVLNode* ANode); //Bool KYAVLTree_Existed(void* ATree, const void* AItem, const void* AData) // { return KYAVLTree_Find(ATree, AItem, AData) != NULL; } // 取 AVL 树的结点 TKYAVLNode* KYAVLTree_Root(void* ATree); // 树的根结点, default: NULL TKYAVLNode* KYAVLTree_Last(void* ATree); // 树的尾结点, default: NULL TKYAVLNode* KYAVLTree_First(void* ATree); // 树的首结点, default: NULL // 取 AVL 结点的属性 long KYAVLTree_Level(TKYAVLNode* ANode); // 结点所在的层号, Level(NULL) = -1 long KYAVLTree_Height(TKYAVLNode* ANode); // 结点的子树高度, Height(NULL) = 0 char KYAVLTree_Balance(TKYAVLNode* ANode); // 结点的子树平衡标志 // 取 AVL 结点的左右子结点及父结点 TKYAVLNode* KYAVLTree_Left(TKYAVLNode* ANode); // 左子结点, Left(NULL) = NULL TKYAVLNode* KYAVLTree_Right(TKYAVLNode* ANode); // 右子结点, Right(NULL) = NULL TKYAVLNode* KYAVLTree_Parent(TKYAVLNode* ANode); // 父结点, Parent(NULL) = NULL // 取 AVL 结点的前后结点 TKYAVLNode* KYAVLTree_Prior(TKYAVLNode* ANode); // 前一结点, Prior(NULL) = NULL TKYAVLNode* KYAVLTree_Next(TKYAVLNode* ANode); // 下一结点, Next(NULL) = NULL #endif

用C语言实现平衡二叉树(AVL tree)源码如下:

// ======================================= // Unit : AVL tree (KYAVLTreeC.c) // Version: 2.1.0.0 (build 2011.03.04) // Author : Kyee Ye // Email : kyee_ye(at)126.com // Copyright (C) Kyee workroom // ======================================= #include <malloc.h> #include "KYAVLTreeC.h" // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ /* 类型定义 */ // 空值 #ifndef NULL #define NULL 0 #endif // 布尔假 #ifndef False #define False 0 #endif // 布尔真 #ifndef True #define True 1 #endif // AVL 树结点项类型 typedef struct _KYAVLItem { TKYAVLNode Node; // 结点 struct _KYAVLItem* Parent; // 父结点项 struct _KYAVLItem* Left; // 左子结点项 struct _KYAVLItem* Right; // 右子结点项 struct _KYAVLItem* Prior; // 前一结点项 struct _KYAVLItem* Next; // 下一结点项 char Balance; // 平衡标志: Height(Left) - Height(Right) Bool Deleting; // 正在删除 } *PKYAVLItem; typedef struct _KYAVLItem TKYAVLItem; // AVL 树类型 typedef struct { void* Data; // 自定义数据 TKYAVLItem* Root; // 根结点 TKYAVLItem* Head; // 首结点 TKYAVLItem* Tail; // 尾结点 long Count; // 结点个数 long MaxCount; // 结点最大个数, 默认值为 0 表示无限制 Bool CanDuplicate; // 是否允许重复, 默认值为 False // 事件 TKYAVL_OnCompare OnCompare; TKYAVL_OnDeletion OnDeletion; } TKYAVLTree, *PKYAVLTree; // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ /* 平衡调整函数 */ // 左右结点调整的平衡增量 static const char _Delta_Balance[2] = {1, -1}; // 调整结点并返回调整后的子树根结点(Left->Left) static TKYAVLItem* L_AdjustLL(TKYAVLTree* ATree, TKYAVLItem* AParent, TKYAVLItem* ATo, TKYAVLItem* ANode) { // 初始化 TKYAVLItem* pGrandparent= AParent->Parent; TKYAVLItem* pRi