题目描述:
哈夫曼树,第一行输入一个数n,表示叶结点的个数。
需要用这些叶结点生成哈夫曼树,根据哈夫曼树的概念,这些结点有权值,即weight,题目需要输出哈夫曼树的带权路径长度(WPL)。
输入格式:
第一行输入一个数n,第二行输入n个叶结点(叶结点权值不超过1000,2<=n<=1000)。
输出格式:
在一行中输出WPL值。
输入样例:
5
1 2 2 5 9
输出样例:
37
参考代码:
#include <>
#define INF 65535
struct huffman{
int w;
int parent,lchild,rchild;
}HT[1001];
int n;
void createHT()
{
int i,j;
//初始化
for(i=0;i<2*n-1;i++)
HT[i].parent=HT[i].lchild=HT[i].rchild=-1;
for(i=0;i<n;i++)
scanf("%d",&HT[i].w);
/*选用两小添新树
删除两小添新人。*/
int a,b;
int a1,b1;
for(i=0;i<n-1;i++)
{
a1=b1=INF;
for(j=0;j<n+i;j++)
{
if(HT[j].parent==-1&&HT[j].w<a1)
{
b=a;
a=j;
b1=a1;
a1=HT[j].w;
}
else if(HT[j].parent==-1&&HT[j].w<b1)
{
b=j;
b1=HT[j].w;
}
}
HT[n+i].w=HT[a].w+HT[b].w;
HT[n+i].lchild=a;
HT[n+i].rchild=b;
HT[a].parent=HT[b].parent=n+i;
}
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
int i,sum=0;
createHT();
for(i=n;i<2*n-1;i++)
sum+=HT[i].w;
printf("%d\n",sum);
}
return 0;
}
参考资料:
传送门