C语言实现哈夫曼树的方法

本文实例为大家分享了C语言实现哈夫曼树的具体代码，供大家参考，具体内容如下

准备工作：

1、定义一个结构体，表示一个节点。其中，这个结构体有4个成员变量，分别表示是这个节点的权值，父节点及左右子节点的下标
2、定义一个整形数组，用于存放各个节点的权值
3、定义一个整形数组，用于存放哈夫曼编码，当然也可以定义一个整形数组来存放哈夫曼编码

构建哈夫曼树：

1、给这个哈夫曼树创建一个结构体数组，其中分配的空间是2 * n - 1，因为我们都知道哈夫曼树的性质有一个是:给定n个叶子节点，那么由这n个叶子节点构成 的哈夫曼树一共含有2 * n - 1个节点。
2、结构体数组前面n个用于存放n个叶子节点，然后后面的n - 1个节点用于存放父节点。这时候，我们需要遍历这个结构体数组，将所有的节点的进行初始化，即节点的权值就是结构体数组各个下标对应的值，然后节点的父节点及子节点的下标为-1，表示的是这个节点没有父节点，同时也没有子节点。
3、遍历数组，将获取数组中两个最小的叶子节点，然后将他们的权值合并构成一个新节点。在这一步中，我们同时需要知道这两个最小节点在结构体数组中的下标，只有这样，我们才可以知道它的父节点的左右子节点的下标，在初始化父节点的时候需要用到。
4、如果已经进行了n - 1次数操作，表明已经构建完成了。

获取哈夫曼编码:

1、由于我们将所有节点的权值都赋值给了一个数组，并且哈夫曼树中的节点的下标和这个数组的下标是一一对应的，那么我们只需要首先在数组中获取这个数字的下标，就表示他在哈夫满树中的下标也是这个，然后往它的父节点方向走，如果当前节点时它父节点的右子节点，那么就将1存放到数组arr2中，否则字符将0存放到数组arr2中。重复这一步，直到当前的节点的父节点为空，及已经遍历到了根节点。
2、重复步骤一，存放数字的数组已经遍历完了，这时候已经将所有数字的哈夫曼编码都已经输出了

#include<stdio.h>
#define MAX_SIZE 1000
typedef struct NODE Node;
struct NODE{
   int weight;//节点的权值
   int parent,lchild,rchild;
};
/*
初始化或者更改节点的信息，比如，如果这个节点是一个新节点，那么
就需要将这个节点初始化成一个叶子节点，否则需要修改这个节点的父节点
*/
void initNode(Node &node,int weight,int parent,int lchild,int rchild){
    node.weight = weight;
    node.lchild = lchild;
    node.rchild = rchild;
    node.parent = parent;
}
/*
1、有n个叶子节点，那么构建哈夫曼树的时候，需要分配n * 2 - 1个内存空间，前n
个表示的是新输入的叶子节点，后面n - 1表示的是叶子节点的父节点
2、遍历这个数组，将进行初始化，即给这些节点的权值赋值，并且将他的左右子节
点、父节点赋初始值为-1，从而构建了n个叶子节点
3、遍历数组，从而将从这个数组中跳出两个值最小的叶子节点，同时需要标记他们的下标，从而可以知道当前最小值节点的父节点的左右子节点的下标，方便下次寻找
最小值的叶子结点的时候不会再找到已经找过的叶子节点
4、将新节点插入到数组的最后。
5、重复3，4的操作，直到只有一个节点，那么这个节点就是哈夫曼树的根节点
*/
void createHuffmanTree(Node *node,int *arr,int n){
    //n表示有n个叶子节点，arr数组存放的是所有叶子节点的值
   int i,j,min1,min2,x1,x2,total;//min1:第一个最小值，min2:第二个最小值,x1:第一个最小值的下标，x2:第二个最小值的下标
   for(i = 0; i < n; i++){
       initNode(node[i],arr[i],-1,-1,-1);//调用initNode方法，从而将节点进行初始化
   }
   total = n * 2 - 1;//total表示的是哈夫曼所有节点的个数
   for(i = n; i < total; i++){
        //i之所以从n开始，是因为第一个父节点这个下标（前n个节点是叶子节点）
        min1 = 65432;//定义两个最小值
        min2 = min1;
        x1 = x2 = 0;//假设两个最小值的下标都是0
        for(j = 0; j < i; j++){
            //判断当前下标的节点是否为叶子节点
           if(node[j].parent == -1){
          //如果当前节点的parent等于-1，表示这个节点是一个叶子节点，那么我们需要判断他是否一个最小节点
                if(node[j].weight < min1){
         //如果当前这个节点的值比min1小，那么我们需要更新min2，min1，同时需要更新两者对应的下标
                    min2 = min1;
                    x2 = x1;
                    min1 = node[j].weight;
                    x1 = j;
                }else if(node[j].weight < min2){
                 /*
                 如果当前这个节点比第一个最小值要大，那么我们需要判断
                 他是否比第二个最小值小，如果是，那么更新min2,并且更新x2
                 */
                   min2 = node[j].weight;
                   x2 = j;
                }
           }
        }
        /*
        找到两个最小节点之后，我们需要将这两个节点的父节点指向i，
        然后将i这个节点进行初始化,并且新节点的左子节点比右子节点小
        从而构建唯一的哈夫曼树
        */
        node[x1].parent = i;
        node[x2].parent = i;
        initNode(node[i],min1 + min2,-1,x1,x2);//初始化合并之后的新节点
   }
}
void huffmanCode(Node *node,int child,int *str){
    //str表示的是这个叶子节点的哈夫曼编码
    int i,parent,j = 0,e;
    parent = node[child].parent;//获取当前这个叶子节点的父节点
    while(parent != -1){
        if(node[parent].lchild == child){
            //如果当前这个节点是父节点的左子节点，那么就将0压入到数组中,否则将1压入数组中
            str[j++] = 0;
        }else{
            str[j++] = 1;
        }
        child = parent;
        parent = node[child].parent;
    }
    e = j;//当退出循环的时候，j表示的是这个数的哈夫曼编码的长度，所以需要从下标为j - 1开始逆序输出，才是这个数的哈夫曼编码
    for(j = e - 1; j>= 0; j--)
        printf("%d",str[j]);
    printf("\n");
}
int main(){
  Node node[MAX_SIZE];
  int arr[MAX_SIZE];//定义一个整形数组，用于存放所有叶子节点的权值
  int arr2[MAX_SIZE];//arr2用于存放一个数字的哈夫曼编码
  int n,i,j,e;
  printf("请输入叶子节点的个数:");
  scanf("%d",&n);
  for(i = 0; i < n; i++)
    scanf("%d",&arr[i]);
  createHuffmanTree(node,arr,n);//构建哈夫曼树
  /*
  获取哈夫曼编码:
  1、将遍历数组arr，从而获得各个叶子节点的权值以及下标
  2、通过这个下标，我们从这个节点向根节点走去，如果当前节点是父节点的左子
  节点，那么将0压入到数组中，否则将1压入到数组arr2中
  3、逆序输出arr2
  */
  for(i = 0; i < n; i++){
    huffmanCode(node,i,arr2);//调用这个方法，将当前下标对应的数字的哈夫曼编码输出
  }
  return 0;
}

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持我们。