1.问题
实例:{5,5,10,10,10,15,20,25}
2.解析
按照贪心算法每次找权值最小(即值最小)的两个数,构成孩子结点,并将这两个数据排除出这一堆数据之外。由于哈弗曼树的数据全在叶子结点,所以把上述权值最小的两个数据构成父结点,然后再将父结点的权值放回上述数据,返回第一步,重复上述过程,直到所有的数据都变成哈夫曼树的叶子结点
3.设计
for循环{
for循环查找无父节点的节点;
O(nlogn)排序;
找出最小的两个节点,插入新的节点作为他们的父节点。
}
4.分析
排序O(nlogn);for循环O(n);插入操作O(logn).
算法时间复杂度是O(nlogn)
5.源代码
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1000;
#define n 8 //叶子数目
#define m 2*n-1 //树中结点总数
typedef struct{ //结点类型
double weight; //结点的权值
int parent,lchild,rchild;//双亲指针及左右孩子
}HTNode;
typedef HTNode HuffmanTree[m];//HuffmanTree是向量类型
typedef struct{ //用于SelectMin函数中排序的结点类型
int id; //保存根结点在向量中的序号
double weight; //保存根结点的权值
}temp;
typedef struct{ //编码结点
char bits[n+1]; //存放编码位串
}CodeNode;
typedef CodeNode HuffmanCode[n];
void InitHuffmanTree(HuffmanTree T){
//初始化哈夫曼树
//将2n-1个结点里的三个指针均置为空(即置为-1),权值置为0
for(int i=0;i<m;i++){
T[i].lchild=-1;
T[i].rchild=-1;
T[i].parent=-1;
T[i].weight=0;
}
}
bool cmp(temp a,temp b){
//用于排序的比较函数
return a.weight<b.weight;
}
void SelectMin(HuffmanTree T,int k,int *p1,int *p2){
//在前k个结点中选择权值最小和次小的根结点,其序号分别为p1和p2
temp x[m]; //x向量为temp类型的向量
int i,j;
for(i=0,j=0;i<=k;i++){ //寻找最小和次小根节点的过程
if(T[i].parent==-1){//如果是根节点,则进行如下操作
x[j].id=i; //将该根节点的序号赋值给x
x[j].weight=T[i].weight;//将该根节点的权值赋值给x
j++; //x向量的指针后移一位
}
}
sort(x,x+j,cmp); //对x按照权值从小到大排序
//排序后的x向量的第一和第二个位置中存储的id是所找的根节点的序号值
*p1=x[0].id;
*p2=x[1].id;
}
void CreateHuffmanTree(HuffmanTree T){
//构造哈夫曼树,T[m-1]为其根结点
int i,p1,p2;
InitHuffmanTree(T); //将T初始化
T[0].weight=5;
T[1].weight=5;
T[2].weight=10;
T[3].weight=10;
T[4].weight=10;
T[5].weight=15;
T[6].weight=20;
T[7].weight=25;
for(i=n;i<m;i++){
//在当前森林T[0..i-1]的所有结点中,选取权最小和次小的
//两个根结点T[p1]和T[p2]作为合并对象
//共进行n-1次合并,新结点依次存于T[i]中
SelectMin(T,i-1,&p1,&p2);//选择权值最小和次小的根结点,其序号分别为p1和p2
//将根为T[p1]和T[p2]的两棵树作为左右子树合并为一棵新的树
//新树的根是新结点T[i]
T[p1].parent=T[p2].parent=i;//T[p1]和T[p2]的两棵树的根结点指向i
T[i].lchild=p1; //最小权的根结点是新结点的左孩子
T[i].rchild=p2; //次小权的根结点是新结点的右孩子
T[i].weight=T[p1].weight+T[p2].weight;//新结点的权值是左右子树的权值之和
}
}
void CharSetHuffmanEncoding(HuffmanTree T,HuffmanCode H){
//根据哈夫曼树T求哈夫曼编码表H
int c,p;//c和p分别指示T中孩子和双亲的位置
char cd[n+1];//临时存放编码
int start;//指示编码在cd中的起始位置
cd[n]='\0';//编码结束符
for(int i=0;i<n;i++){//依次求叶子T[i]的编码
start=n;//编码起始位置的初值
c=i;//从叶子T[i]开始上溯
while((p=T[c].parent)>=0){//直至上溯到T[c]是树根为止
//若T[c]是T[p]的左孩子,则生成代码0;否则生成代码1
if(T[p].lchild==c)
cd[--start]='0';
else
cd[--start]='1';
c=p;//继续上溯
}
strcpy(H[i].bits,&cd[start]);//复制编码位串
}
}
int main(){
HuffmanTree T;
HuffmanCode H;
CreateHuffmanTree(T);
CharSetHuffmanEncoding(T,H);
printf("哈夫曼编码:\n");
double wpl=0.0;
for(int i=0;i<n;i++){
printf("id:%d value:%.0f\n",i,T[i].weight);
wpl+=strlen(H[i].bits)*T[i].weight;
}
printf("平均码长为:%.2lf\n",wpl/100);
return 0;
}