哈夫曼树使用c++编码
对哈夫曼树按中序遍历输出对应叶子的哈夫曼编码
#include<string.h>
#include<iostream>
using namespace std;
#define N 50 //叶子结点数
#define M 2*N-1 //树中结点总数
typedef struct node {
int weight; //权重
int parent; //双亲结点
int lchild; //左孩子结点
int rchild; //右孩子结点
}huffnode;
typedef struct list{
char cd[N]; //存放哈夫曼码
int begin;
}huffcode;
void createhuff(huffnode ht[],int n){ //构造哈夫曼树
int lnode,rnode;
int min1,min2;
for(int i=0;i<2*n-1;i++){ //所有结点赋初值
ht[i].parent=-1;
ht[i].lchild=-1;
ht[i].rchild=-1;
}
for(int i=n;i<2*n-1;i++){ //构造哈夫曼树
min1=min2=1024; //lchild和rchild为最小权重的两个结点位置
lnode=rnode=-1;
for(int k=0;k<=i-1;k++){
if(ht[k].parent==-1){ //只在尚未构造二叉树的结点中查找
if(ht[k].weight<min1){
min2=min1;
rnode=lnode;
min1=ht[k].weight;
lnode=k;
}
else if(ht[k].weight<min2){
min2=ht[k].weight;
rnode=k;
}
}
}
ht[lnode].parent=i; ht[rnode].parent=i; ht[i].weight=ht[lnode].weight+ht[rnode].weight;
ht[i].lchild=lnode; ht[i].rchild=rnode;
}
}
void createhuffcode(huffnode ht[],huffcode hcd[],int n){ //构造哈夫曼编码
int i,f,c;
huffcode hc;
for(i=0;i<n;i++){ //根据哈夫曼树求哈夫曼编码
hc.start=n; c=i; f=ht[i].parent;
while(f!=-1){ //循序直到树根结点
if(ht[f].lchild==c) //处理左孩子结点
hc.cd[hc.begin--]='0';
else
hc.cd[hc.begin--]='1'; //处理右孩子结点
c=f;
f=ht[f].parent; }
hc.begin++; //start指向哈夫曼编码最开始字符
hcd[i]=hc; }}
void inorder(huffnode ht[],int array[],int i){ //对哈夫曼树进行中序遍历,并将树存入新数组中
static int j=0; //设置静态变量
if(ht[i].lchild!=-1){
inorder(ht,array,ht[i].lchild);
if(ht[i].lchild==-1){
array[j]=i;
j++;
}
inorder(ht,array,ht[i].rchild);
}
else{
array[j]=i;
j++;
}
}
void printhcode(huffnode ht[],huffcode hcd[],int n,int array[]){ // 输出哈夫曼编码
int i,k,m;
for(m=0;m<n;m++){
i=array[m];
cout<<ht[i].weight<<" ";
for(k=hcd[i].begin;k<=n;k++){
cout<<hcd[i].cd[k];
}
cout<<endl;
}
}
int main(){
int n,i;
huffnode ht[M];
huffcode hcd[N];
int num[80];
cin>>n;
int array[n-1];
for(i=0;i<n;i++){
cin>>num[i];
}
for(i=0;i<n;i++)
{ ht[i].weight=num[i]; }
createhuff(ht,n); //创建哈夫曼树
createhuffcode(ht,hcd,n); //对哈夫曼进行编码
inorder(ht,array,2*n-2); //进行中序输出
printhcode(ht,hcd,n,array); //输出哈夫曼编码
return 0 ;
}