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

#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;
}