数据结构的实践心得(二叉树的抽象数据结构和各种遍历的递归及非递归实现,以及哈夫曼算法应用:binaryTree)
头文件(binaryTree.h):
#pragma once
#define BinaryTreeOrderBUFFER 128
typedef double binaryElemType;
typedef struct
{
int key; // 二叉树节点关键字(优先数)
binaryElemType *data; // 数据内容
int dataSize; // 数据内容的空间大小
struct binaryNode *parent; // 父节点
struct binaryNode *left; // 左节点
struct binaryNode *right; // 右节点
int length; // 二叉树的节点数量(包括所有子节点)
int isRight; // 如果是子节点,则是否为右节点(右:1;左:0)
}binaryNode;
// 创建二叉树节点
binaryNode* createBinaryNode(int dataSize);
// 设置二叉树节点的数据内容
void setBinaryNodeData(binaryNode *T, binaryElemType *e);
// 是否为叶子节点返回1,否则返回0
int isLeafBinaryNode(binaryNode *T);
// 二叉树的节点数量
int lengthBinaryList(binaryNode *T);
// 二叉树的节点数量(递归)
int lengthBinaryListRecursion(binaryNode *T);
// 取得二叉树节点的层级
int levelBinaryNode(binaryNode *T);
// 取得子节点的相对层级
int levelBinaryChildNode(binaryNode *T, binaryNode *child);
// 取得二叉树节点的根节点
binaryNode* getBinaryRoot(binaryNode *T);
// 插入二叉树的子节点(右:1;左:0)
binaryNode* insertBinaryNode(binaryNode *T, int isRight, int key);
// 添加二叉树的子节点(右:1;左:0)
binaryNode* addBinaryNode(binaryNode *T, int isRight, binaryNode *child);
// 移除二叉树的子节点
int removeBinaryNode(binaryNode *T, binaryNode *child);
// 清空二叉树列表(递归)
void clearBinaryList(binaryNode *T);
// 层次遍历(输出节点的排序数组)
int levelOrder(binaryNode *T, binaryNode **orderList);
// 前序遍历(递归)
int preOrderRecursion(binaryNode *T, binaryNode **orderList);
// 前序遍历(输出数据的排序数组)
int preOrder(binaryNode *T, binaryNode **orderList);
// 中序遍历(递归)
int middleOrderRecursion(binaryNode *T, binaryNode **orderList);
// 中序遍历(输出数据的排序数组)
int middleOrder(binaryNode *T, binaryNode **orderList);
// 后序遍历(递归)
int postOrderRecursion(binaryNode *T, binaryNode **orderList);
// 后序遍历(输出数据的排序数组)
int postOrder(binaryNode *T, binaryNode **orderList);
// 生成最优二叉树(哈夫曼算法),返回二叉树根节点,参数传出叶子节点关系
binaryNode* bestBinaryTree(binaryNode *dataListOutleaf, int dataLen);
// 取得哈夫曼编码(叶子节点列表),注意在外部释放每个hfCode的内存空间
char** getHuffmanCodeByLeaf(binaryNode *leafList, int leafLen, char **hfCodeList);
// 取得哈夫曼编码,注意在外部释放每个hfCode的内存空间
char** getHuffmanCode(binaryNode *dataList, int dataLen, char **hfCodeList);
程序文件(binaryTree.c):
#include <stdlib.h>
#include "mystring.h"
#include "linkQueue.h"
#include "linkStack.h"
#include "binaryTree.h"
// 创建二叉树节点
binaryNode* createBinaryNode(int dataSize)
{
// 首先申请自己的内存空间
binaryNode *bn = (binaryNode *)malloc(sizeof(binaryNode));
if (!bn) return NULL;
if (dataSize > 0)
{
bn->data = (binaryElemType *)malloc(sizeof(binaryElemType) * dataSize);
if (!bn->data)
{
// 清空二叉树列表(递归)
clearBinaryList(bn);
return NULL;
}
bn->dataSize = dataSize;
}
else
{
bn->data = NULL;
bn->dataSize = 0;
}
bn->key = 0;
bn->parent = NULL;
bn->left = NULL;
bn->right = NULL;
bn->length = 1;
bn->isRight = 0;
return bn;
}
// 设置二叉树节点的数据内容
void setBinaryNodeData(binaryNode *T, binaryElemType *data)
{
if ((!T) || (!data)) return;
// 判断数据内容是否有效
if (T->data)
{
// 是否包含数据内容
int i;
// 设置数据内容
for (i = 0; i < T->dataSize; i++)
{
T->data[i] = data[i];
}
}
}
// 是否为叶子节点返回1,否则返回0
int isLeafBinaryNode(binaryNode *T)
{
if (!T) return -1;
return ((!T->left) && (!T->right));
}
// 二叉树的节点数量
int lengthBinaryList(binaryNode *T)
{
if (!T) return 0;
// 判断节点数量是否有效
if (T->length > 0)
{
return T->length;
}
else
{
// 二叉树的节点数量(递归)
return lengthBinaryListRecursion(T);
}
}
// 二叉树的节点数量(递归)
int lengthBinaryListRecursion(binaryNode *T)
{
if (!T) return 0;
// 二叉树的节点数量
int len = 1;
// 递归记录左子树数量
len += lengthBinaryList(T->left);
// 递归记录右子树数量
len += lengthBinaryList(T->right);
// 更新节点数量
T->length = len;
return len;
}
// 取得二叉树节点的层级
int levelBinaryNode(binaryNode *T)
{
if (!T) return -1;
// 从 0 开始
int level = 0;
binaryNode *bn = T;
// 判断父级节点
while (bn->parent)
{
level++;
// 递归父级节点
bn = bn->parent;
}
return level;
}
// 取得子节点的相对层级
int levelBinaryChildNode(binaryNode *T, binaryNode *child)
{
if ((!T) || (!child)) return -1;
// 从 0 开始
int level = 0;
binaryNode *bn = child;
// 判断节点
while (bn)
{
if (T == bn) return level;
level++;
// 递归父级节点
bn = bn->parent;
}
return -1;
}
// 取得二叉树节点的根节点
binaryNode* getBinaryRoot(binaryNode *T)
{
if (!T) return NULL;
binaryNode *bn = T;
// 判断父级节点
while (bn->parent)
{
// 递归父级节点
bn = bn->parent;
}
return bn;
}
// 插入二叉树的子节点(右:1;左:0)
binaryNode* insertBinaryNode(binaryNode *T, int isRight, int key)
{
if (!T) return NULL;
// 判断是否已有左节点或右节点
if (isRight > 0)
{
if (T->right) return NULL;
}
else
{
if (T->left) return NULL;
}
// 创建二叉树节点
binaryNode *child = createBinaryNode(T->dataSize);
if (!child) return NULL;
// 节点关键字(优先数)
child->key = key;
// 父级节点
child->parent = T;
// 判断插入左节点或右节点
if (isRight > 0)
{
T->right = child;
child->isRight = 1;
}
else
{
T->left = child;
child->isRight = 0;
}
// 维护节点数量
binaryNode *bn = T;
// 判断节点
while (bn)
{
bn->length++;
// 递归父级节点
bn = bn->parent;
}
return child;
}
// 添加二叉树的子节点(右:1;左:0)
binaryNode* addBinaryNode(binaryNode *T, int isRight, binaryNode *child)
{
if ((!T) || (!child)) return NULL;
// 判断是否已有左节点或右节点
if (isRight > 0)
{
if (T->right) return NULL;
}
else
{
if (T->left) return NULL;
}
// 避免嵌套死循环,相对层级 >= 0,则表示嵌套子级节点
if (levelBinaryChildNode(child, T) >= 0) return NULL;
// 移除原有的父级关系
removeBinaryNode(child->parent, child);
child->parent = T;
// 判断添加左节点或右节点
if (isRight > 0)
{
T->right = child;
child->isRight = 1;
}
else
{
T->left = child;
child->isRight = 0;
}
// 维护节点数量
binaryNode *bn = T;
// 判断节点
while (bn)
{
bn->length += child->length;
// 递归父级节点
bn = bn->parent;
}
return child;
}
// 移除二叉树的子节点
int removeBinaryNode(binaryNode *T, binaryNode *child)
{
if ((!T) || (!child)) return -1;
// 判断父级关系是否匹配
if (T != child->parent) return -1;
int result = -1;
// 判断是否移除左节点
if (T->left == child)
{
T->left = NULL;
result = 1;
}
// 判断是否移除右节点
if (T->right == child)
{
T->right = NULL;
result = 1;
}
if (result > 0)
{
child->parent = NULL;
// 维护节点数量
binaryNode *bn = T;
// 判断节点
while (bn)
{
bn->length -= child->length;
// 递归父级节点
bn = bn->parent;
}
}
return result;
}
// 清空二叉树列表(递归)
void clearBinaryList(binaryNode *T)
{
if (!T) return;
// 递归清空左节点
clearBinaryList(T->left);
// 递归清空右节点
clearBinaryList(T->right);
// 释放数据空间
if (T->data) free(T->data);
// 释放自己
free(T);
}
// 层次遍历(输出数据的排序数组)
int levelOrder(binaryNode *T, binaryNode **orderList)
{
if ((!T) || (!orderList)) return -1;
linkQueue que;
// 初始化队列
initializeLinkQueue(&que);
// 根节点进队
pushQueue(&que, (int)T);
int index = 0;
binaryNode *bn;
// 进行队列操作
while (!isEmptyLinkQueue(que))
{
// 取得队首元素
bn = (int)front(que);
// 将队首元素出队
popQueue(&que);
// 判断元素是否有效
if (!bn) break;
// 排序数组赋值
orderList[index++] = bn;
// 左节点进队
if (bn->left) pushQueue(&que, (int)bn->left);
// 右节点进队
if (bn->right) pushQueue(&que, (int)bn->right);
}
// 清空队列
clearLinkQueue(&que);
return index;
}
// 前序遍历(递归)
int preOrderRecur(binaryNode *T, binaryNode **orderList, int index)
{
int i = index;
// 排序数组赋值
orderList[i++] = T;
// 递归左节点
if (T->left) i = preOrderRecur(T->left, orderList, i);
// 递归左节点
if (T->right) i = preOrderRecur(T->right, orderList, i);
return i;
}
// 前序遍历(递归)
int preOrderRecursion(binaryNode *T, binaryNode **orderList)
{
if ((!T) || (!orderList)) return -1;
// 前序遍历(递归)
return preOrderRecur(T, orderList, 0);
}
// 前序遍历(输出数据的排序数组)
int preOrder(binaryNode *T, binaryNode **orderList)
{
if ((!T) || (!orderList)) return -1;
linkStack stack;
// 初始化栈
initializeLinkStackSeq(&stack, BinaryTreeOrderBUFFER);
// 根节点进栈
push(&stack, (int)T);
int index = 0;
binaryNode *bn;
// 进行栈操作
while (!isEmptyLinkStack(stack))
{
// 取得栈首元素
bn = (int)top(stack);
// 将栈首元素出栈
pop(&stack);
// 判断元素是否有效
if (!bn) break;
// 排序数组赋值
orderList[index++] = bn;
// 右节点进栈
if (bn->right) push(&stack, (int)bn->right);
// 左节点进栈
if (bn->left) push(&stack, (int)bn->left);
}
// 清空栈
clearLinkStack(&stack);
return index;
}
// 中序遍历(递归)
int middleOrderRecur(binaryNode *T, binaryNode **orderList, int index)
{
int i = index;
// 递归左节点
if (T->left) i = middleOrderRecur(T->left, orderList, i);
// 排序数组赋值
orderList[i++] = T;
// 递归左节点
if (T->right) i = middleOrderRecur(T->right, orderList, i);
return i;
}
// 中序遍历(递归)
int middleOrderRecursion(binaryNode *T, binaryNode **orderList)
{
if ((!T) || (!orderList)) return -1;
// 中序遍历(递归)
return middleOrderRecur(T, orderList, 0);
}
// 中序遍历(输出数据的排序数组)
int middleOrder(binaryNode *T, binaryNode **orderList)
{
if ((!T) || (!orderList)) return -1;
linkStack stack, stackStatus;
// 初始化栈
initializeLinkStackSeq(&stack, BinaryTreeOrderBUFFER);
initializeLinkStackSeq(&stackStatus, BinaryTreeOrderBUFFER);
// 根节点进栈
push(&stack, (int)T);
push(&stackStatus, 0);
int index = 0, status;
binaryNode *bn;
// 进行栈操作
while (!isEmptyLinkStack(stack))
{
// 取得栈首元素
bn = (int)top(stack);
// 判断元素是否有效
if (!bn) break;
// 取得访问状态
status = (int)top(stackStatus);
pop(&stackStatus);
// 先访问左节点
if (status == 0)
{
push(&stackStatus, 1);
// 标记为0,尝试压栈左节点
if (bn->left)
{
push(&stack, (int)bn->left);
push(&stackStatus, 0);
}
}
else
{
// 将栈首元素出栈
pop(&stack);
// 排序数组赋值
orderList[index++] = bn;
// 尝试压栈右节点
if (bn->right)
{
push(&stack, (int)bn->right);
push(&stackStatus, 0);
}
}
}
// 清空栈
clearLinkStack(&stack);
clearLinkStack(&stackStatus);
return index;
}
// 后序遍历(递归)
int postOrderRecur(binaryNode *T, binaryNode **orderList, int index)
{
int i = index;
// 递归左节点
if (T->left) i = postOrderRecur(T->left, orderList, i);
// 递归左节点
if (T->right) i = postOrderRecur(T->right, orderList, i);
// 排序数组赋值
orderList[i++] = T;
return i;
}
// 后序遍历(递归)
int postOrderRecursion(binaryNode *T, binaryNode **orderList)
{
if ((!T) || (!orderList)) return -1;
// 后序遍历(递归)
return postOrderRecur(T, orderList, 0);
}
// 后序遍历(输出数据的排序数组)
int postOrder(binaryNode *T, binaryNode **orderList)
{
if ((!T) || (!orderList)) return -1;
linkStack stack, stackStatus;
// 初始化栈
initializeLinkStackSeq(&stack, BinaryTreeOrderBUFFER);
initializeLinkStackSeq(&stackStatus, BinaryTreeOrderBUFFER);
// 根节点进栈
push(&stack, (int)T);
push(&stackStatus, 0);
int index = 0, status;
binaryNode *bn;
// 进行栈操作
while (!isEmptyLinkStack(stack))
{
// 取得栈首元素
bn = (int)top(stack);
// 判断元素是否有效
if (!bn) break;
// 取得访问状态
status = (int)top(stackStatus);
pop(&stackStatus);
// 先左后右访问最低层节点
if (status == 0)
{
push(&stackStatus, 1);
// 标记为0,尝试压栈左节点
if (bn->left)
{
push(&stack, (int)bn->left);
push(&stackStatus, 0);
}
}
else if (status == 1)
{
push(&stackStatus, 2);
// 标记为1,尝试压栈右节点
if (bn->right)
{
push(&stack, (int)bn->right);
push(&stackStatus, 0);
}
}
else
{
// 将栈首元素出栈
pop(&stack);
// 排序数组赋值
orderList[index++] = bn;
}
}
// 清空栈
clearLinkStack(&stack);
clearLinkStack(&stackStatus);
return index;
}
// 在所有父级节点为空的元素中查找Key值(优先数)最小节点的索引下标
int minKeyIndex(binaryNode **nodeList, int len, int curIndex)
{
int i, min = -1;
binaryNode *bn;
// 倒序遍历已创建的二叉树节点列表
for (i = len - 1; i >= curIndex; i--)
{
bn = nodeList[i];
// 已设置父级节点,则无需比较
if (bn->parent) continue;
// 初始化最小值索引
if (min < 0) min = i;
// 比较Key值(优先数)最小节点的索引下标
if (bn->key < nodeList[min]->key) min = i;
}
return min;
}
// 生成最优二叉树(哈夫曼算法),返回二叉树根节点,参数传出叶子节点关系
binaryNode* bestBinaryTree(binaryNode *dataListOutleaf, int dataLen)
{
if ((!dataListOutleaf) || (dataLen < 2)) return NULL;
// 构造的最优二叉树,总数为叶子节点的两倍减一(dataLen * 2 - 1)
int len = dataLen * 2 - 1;
// 创建二叉树节点的缓存列表
binaryNode **nodeList = (binaryNode **)malloc(sizeof(binaryNode *) * len);
if (!nodeList) return NULL;
// 数据内容空间大小
int dataSize = dataListOutleaf[0].dataSize;
binaryNode *bn;
int index = len - 1, k;
// 循环创建二叉树节点,并倒序存储在列表中
for (k = 0; k < dataLen; k++)
{
// 创建二叉树节点
bn = createBinaryNode(dataSize);
if (!bn) continue;
// 优先数赋值
bn->key = dataListOutleaf[k].key;
// 设置二叉树节点的数据内容
setBinaryNodeData(bn, dataListOutleaf[k].data);
// 在列表中保存二叉树节点
nodeList[index--] = bn;
}
int minLeft, minRight, leafIndex;
binaryNode *leftNode, *rightNode;
// 数组左侧尚有未用空间,即新创建的节点个数还不足
while (index >= 0)
{
// 创建二叉树节点
bn = createBinaryNode(0);
if (!bn) continue;
// 在所有父级节点为空的元素中查找最小值的索引下标
minLeft = minKeyIndex(nodeList, len, index + 1);
leftNode = nodeList[minLeft];
// 添加二叉树的子节点(右:1;左:0)
addBinaryNode(bn, 0, leftNode);
// 如果是叶子节点,更新传入的基础数据列表的父级指针
if (isLeafBinaryNode(leftNode))
{
leafIndex = len - minLeft - 1;
dataListOutleaf[leafIndex].parent = bn;
dataListOutleaf[leafIndex].isRight = 0;
}
// 在所有父级节点为空的元素中查找最小值的索引下标
minRight = minKeyIndex(nodeList, len, index + 1);
rightNode = nodeList[minRight];
// 添加二叉树的子节点(右:1;左:0)
addBinaryNode(bn, 1, rightNode);
// 如果是叶子节点,更新传入的基础数据列表的父级指针
if (isLeafBinaryNode(rightNode))
{
leafIndex = len - minRight - 1;
dataListOutleaf[leafIndex].parent = bn;
dataListOutleaf[leafIndex].isRight = 1;
}
// 新创建的节点Key值(优先数)
bn->key = (leftNode->key + rightNode->key);
// 在列表中保存二叉树节点
nodeList[index--] = bn;
}
// 释放缓存列表
free(nodeList);
return bn;
}
// 取得哈夫曼编码(叶子节点列表),注意在外部释放每个hfCode的内存空间
char** getHuffmanCodeByLeaf(binaryNode *leafList, int leafLen, char **hfCodeList)
{
if ((!leafList) || (!hfCodeList) || (leafLen < 2)) return NULL;
// 定义一个字符串缓存
char *tmp = (char *)malloc(sizeof(char) * leafLen);
if (!tmp) return NULL;
binaryNode *bn;
char *code;
int i, index;
// 为每个hfCode申请内存空间
for (i = 0; i < leafLen; i++)
{
code = (char *)malloc(sizeof(char) * leafLen);
if (!code) return NULL;
bn = &(leafList[i]);
index = 0;
// 开始记录编码
while (bn->parent)
{
tmp[index++] = bn->isRight + '0';
// 递归父级节点
bn = bn->parent;
}
// 补充结束符
tmp[index] = 0;
// 字符串逆序输出
myReverse(tmp, code);
// 保存hfCode
hfCodeList[i] = code;
}
// 释放缓存
free(tmp);
return hfCodeList;
}
// 取得哈夫曼编码,注意在外部释放每个hfCode的内存空间
char** getHuffmanCode(binaryNode *dataList, int dataLen, char **hfCodeList)
{
// 生成最优二叉树(哈夫曼算法),返回二叉树根节点,参数传出叶子节点关系
binaryNode *root = bestBinaryTree(dataList, dataLen);
if (!root) return NULL;
// 取得哈夫曼编码(叶子节点列表)
char **result = getHuffmanCodeByLeaf(dataList, dataLen, hfCodeList);
// 释放二叉树列表
clearBinaryList(root);
return result;
}