哈夫曼树

声明：图片及内容基于https://www.bilibili.com/video/BV1MK411j7CR?from=articleDetail

相关概念

特点

构造过程

存储结构

代码

#include<iostream> #include<vector> #include <limits.h> #include<iomanip>
using namespace std; struct element { int weight; int lchild, rchild, parent; }; void select(element huffTree[], int n, int& s1, int& s2);  //传递引用 void HuffmanTree(element huffTree[], int w[], int n) { for (int i = 0; i < 2 * n - 1; i++) {     //初始化哈夫曼树 注意2*n-1给结点
        huffTree[i].lchild = -1; huffTree[i].parent = -1; huffTree[i].rchild = -1; } for (int i = 0; i < n; i++) { huffTree[i].weight = w[i];  //复制权值 
 } for (int k = n; k < 2 * n - 1; k++) { int i1, i2;          //i1最小，i2次小 为下标 
        select(huffTree, k, i1, i2); //注意这里必须传k，k是个变量，代表已经有k个结点作为待选数据，且每次k都增加 huffTree[k].weight = huffTree[i1].weight + huffTree[i2].weight; huffTree[i1].parent = k; huffTree[i2].parent = k; huffTree[k].lchild = i1; huffTree[k].rchild = i2; cout << "最小下标：" << i1 << " 次小下标：" << i2 << endl; } } void select(element huffTree[], int n, int& s1, int& s2) {  //传递引用 int i; s1 = s2 = 0; int min1 = INT_MAX;//最小值，INT_MAX在<limits.h>中定义的，INT_MAX是最大值2^31-1
    int min2 = INT_MAX;//次小值，找最小值就默认初始值为最大值

    for (i = 0; i < n; ++i) { if (huffTree[i].parent == -1) {//筛选没有父节点的最小和次小权值下标
            if (huffTree[i].weight < min1) {//如果比最小值小
                min2 = min1; //min1不是最小值了，成为次小值，赋给min2 s2 = s1; min1 = huffTree[i].weight; //把新的最小值赋给min1 s1 = i; } else if ((huffTree[i].weight >= min1) && (huffTree[i].weight < min2)) {//如果大于等于最小值，且小于次小值
                min2 = huffTree[i].weight; //更新最小值 s2 = i; } else {//如果大于次小值，则什么都不做
 ; } } } } int main() { int n;   //结点个数
    cout << "请输入结点个数：" << endl; cin >> n; int w[100]; cout << "请输入" << n << "个权值" << endl; for (int i = 0; i < n; i++) {     //初始化权值 
        int t; cin >> t; w[i] = t; } struct element huffTree[100]; HuffmanTree(huffTree, w, n); cout << "打印哈夫曼树的数组内容："<<endl; cout << "weight parent lchild rchild" << endl; for (int i = 0; i < 2*n-1; i++) { cout << setw(2) << huffTree[i].weight << "      " << setw(2) << huffTree[i].parent << "      " << setw(2) << huffTree[i].lchild << "      " << setw(2) << huffTree[i].rchild << endl; } return 0; }

输入：

4

2 4 5 3

输出：

最小下标：0 次小下标：3
最小下标：1 次小下标：2
最小下标：4 次小下标：5
打印哈夫曼树的数组内容：
weight parent lchild rchild
2 4 -1 -1
4 5 -1 -1
5 5 -1 -1
3 4 -1 -1
5 6 0 3
9 6 1 2
14 -1 4 5