哈夫曼树带权路径长度求解(C语言实现)

时间:2025-03-14 07:52:29

题目描述:
哈夫曼树,第一行输入一个数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;
} 

参考资料:
传送门