代码:
#include<stdio.h>
#define ERROR 0
#define OK 1
typedef int Status;
//采用顺序存储结构,一维结构数组
//定义结点类型
typedef struct
{
int weight;
int parent,lch,rch;
}HTNode,*HaffmanTree;
HaffmanTree HT;
//哈夫曼树的构造算法
Status CreatHT(HaffmanTree HT,int n)
{
int s1,s2;
if(n<=1)
return ERROR;
int m=2*n-1;//数组共2*n-1个元素
int i;
HT=new HTNode[m+1];//构建定义好的结构体类型数组
//初始化数组
for(i=1;i<=m;i++) //0号下标没有用,HT[m]表示元素的根结点
{
HT[i].lch=0;//将2n-1个元素的lch,rch,parent置为0
HT[i].rch=0;
HT[i].parent=0;
}
for(i=1;i<=n;i++)
{
scanf("%d",&HT[i].weight);//输入前n个元素的weight值
}
//建立哈夫曼树
for(i=n+1;i<=m;i++)//合并产生n-1个新结点
{
void Select(HaffmanTree HT,int k,int &s1,int &s2);
//在HT[k](1<k<i-1)中选择两个双亲域为0,并且最小的结点,并返回它们在HT中的序号
Select(HT,i-1,s1,s2);
HT[s1].parent=i;
HT[s2].parent=i;//表示从F中删除s1,s2
HT[i].lch=s1;
HT[i].rch=s2;//s1,s2分别作为i的左右孩子
HT[i].weight=HT[s1].weight+HT[s2].weight;//i的权值为左右孩子的权值之和
}
return OK;
}
void Select(HaffmanTree HT,int k,int &s1,int &s2)
{
int i,j;
int min1,min2;
for(i=1;i<k;i++)
{
if(HT[i].parent==0)
{
min1=HT[1].weight;
s1=1;
if(HT[i].weight<min1)
{
int t;
t=min1;
min1=HT[i].weight;
HT[i].weight=t;//选出权值最小的元素
s1=i;
}
min2=HT[1].weight;
s2=1;
if(HT[i].weight<min2&&HT[i].weight>min1)
{
int x;
x=min2;
min2=HT[i].weight;
HT[i].weight=x;//选出权值次小的元素
s2=i;
}
}
}
}
//主函数测试
int main()
{
int n;
printf("请输入要构造成哈夫曼树的结点数:\n");
scanf("%d",&n);
printf("请输入各结点所对应的权重:\n");
if(CreatHT(HT,n))
printf("CreatHT success\n");
return 0;
}