哈夫曼树

3.7哈夫曼树

定义：具有最小带权路径长度的二叉树称为哈夫曼树，又称为最优二叉树。

哈夫曼树的构造原则：

权值越大的节点越靠近根节点，权值越小的越远离根节点。

由以上规则，我们可以得出构造二叉树的基本过程：

首先对给出的数据进行排序，接着我们拿出最小的两个数据构成二叉树，该二叉树的根节点是这两个节点的和，同时将本次得到的和放入之前的数据中重新进行排序，再重复上述步骤得到哈夫曼树。

3.7哈夫曼树

定义：具有最小带权路径长度的二叉树称为哈夫曼树，又称为最优二叉树。

哈夫曼树的构造原则：

权值越大的节点越靠近根节点，权值越小的越远离根节点。

由以上规则，我们可以得出构造二叉树的基本过程：

首先对给出的数据进行排序，接着我们拿出最小的两个数据构成二叉树，该二叉树的根节点是这两个节点的和，同时将本次得到的和放入之前的数据中重新进行排序，再重复上述步骤得到哈夫曼树。

代码：

#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
typedef struct TreeNode{

  int data;
  int parent,ltree,rtree;

}TreeNode;
typedef struct Huffmantree{

  TreeNode * data;
  int length;
}Huffmantree;
Huffmantree * Inittree(int * weight,int length)//初始化哈夫曼树
{
  //cout<<length<<endl;
  Huffmantree * tree=(Huffmantree *)malloc(sizeof(Huffmantree));
  tree->data=(TreeNode *)malloc(sizeof(TreeNode)*(2*length-1));
  tree->length=length;
  for(int i=0;i<length;i++)
  {
    tree->data[i].data=weight[i];
    tree->data[i].parent=0;
    tree->data[i].ltree=-1;
    tree->data[i].rtree=-1;

  }
  return tree;
}
int * Selectnode(Huffmantree * tree)//找到最小的两个节点
{
  int min1=9999;
  int min2=9999;
  int node1,node2;
  for(int i=0;i<tree->length;i++)
  {
    if(tree->data[i].parent==0&&tree->data[i].data<min1)
    {
      min1=tree->data[i].data;
       node1=i;
    }
  }
  for(int i=0;i<tree->length;i++)
  {
    if(tree->data[i].parent==0&&tree->data[i].data<min2)
    {
      if(i!=node1)
      {
        min2=tree->data[i].data;
        node2=i;
      }

    }

  }
  int *res=(int *)malloc(sizeof(int)*2);
  res[0]=node1;
  res[1]=node2;
 // cout<<res[0]<<' '<<node2<<endl;
  return res;
}
void CreateHuffmantree(Huffmantree * tree)
{
  int length=tree->length*2-1;
  int ndoe1,node2;
  for(int i=tree->length;i<length;i++)
  {
    int * res=Selectnode(tree);
    tree->data[i].data=tree->data[res[0]].data+tree->data[res[1]].data;
    tree->data[i].ltree=res[0];
    tree->data[i].rtree=res[1];
    tree->data[i].parent=0;
    tree->data[res[0]].parent=i;
    tree->data[res[1]].parent=i;
    tree->length++;//因为加了一个节点长度+1
  }
  //cout<<tree->length;
}

void pre_print_Huffmantree(Huffmantree * tree,int index)
{
  if(index!=-1)
  {
    cout<<tree->data[index].data<<" ";
    pre_print_Huffmantree(tree,tree->data[index].ltree);
    pre_print_Huffmantree(tree,tree->data[index].rtree);
  }
}
int main()
{
  int weight[1001];
  int n;
  cin>>n;
  for(int i=0;i<n;i++)
  {
    cin>>weight[i];
  }
  Huffmantree * HF=Inittree(weight,n);
  CreateHuffmantree(HF);
  //cout<<HF->length-1<<endl;
  pre_print_Huffmantree(HF,HF->length-1);
  //Selectnode(HF,node1,node2);
  return 0;
}