【问题描述】
已知输入一串正整数,正整数之间用空格键分开,请建立一个哈夫曼树,以输入的数字为叶节点,求这棵哈夫曼树的带权路径长度。
【输入形式】
首先输入正整数的个数,然后接下来为接下来的正整数,正整数个数不超过10个
【输出形式】
输出相应的权值
【样例输入】
5 4 5 6 7 8
【样例输出】
69
【样例说明】
【评分标准】
#include<stdio.h>
#include<stdlib.h>
typedef struct
{
unsigned int weight;
unsigned int parent,lchild,rchild;
}HTNode,*HuffmanTree;
void Select(HuffmanTree HT,int n,int &s1,int &s2)
{
for(int i=1;i<=n;i++)
{
if(HT[i].parent==0)
{ s1=i;
break;
}
}
for(int i=1;i<=n;i++)
{
if(HT[i].parent==0&&HT[i].weight<=HT[s1].weight)
s1=i;
}
for(int i=1;i<=n;i++)
{
if(HT[i].parent==0&&i!=s1)
{ s2=i;
break;
}
}
for(int i=1;i<=n;i++)
{
if(HT[i].parent==0&&HT[i].weight<=HT[s2].weight&&i!=s1)
s2=i;
}
}
void HuffmanCoding(HuffmanTree &HT,int *w,int n)
{
if(n<=1) return;
int m=2*n-1;
HT=(HuffmanTree)malloc((m+1)*sizeof(HTNode));
HuffmanTree p=HT+1;
int i,s1,s2;
for(i=1;i<=n;i++,p++ ,w++) //前n个节点放n个叶子
{
p->weight=*w;
p->parent=0;
p->rchild=0;
p->lchild=0;
}
for(;i<=m;i++,p++) //后面n-1放有孩子的节点
{
p->weight=0;
p->parent=0;
p->rchild=0;
p->lchild=0;
}
for(i=n+1;i<=m;i++) //建立哈夫曼树
{
Select(HT,i-1,s1,s2);
//printf("%d %d\n",s1,s2);
HT[s1].parent=i;
HT[s2].parent=i;
HT[i].lchild=s1;
HT[i].rchild=s2;
HT[i].weight=HT[s1].weight+HT[s2].weight;
}
}
int WeightLength(HuffmanTree HT,int n) //n个叶子
{
int c,s=0;//c是路径长度,s是带权路径长度
for(int i=1;i<=n;i++)
{
c=0;
int k=i;
while(HT[k].parent!=0)
{
k=HT[k].parent;
c++;
}
s=s+c*HT[i].weight;
}
return s;
}
int main()
{
int n;
HuffmanTree HT;
scanf("%d",&n);
int w[100];
for(int i=0;i<n;i++)
scanf("%d",&w[i]);
HuffmanCoding(HT,w,n);
printf("%d",WeightLength(HT,n));
return 0;
}