题目:
输入一个整数n,代表样例个数
输入一个数m,表示叶结点的个数,接下来输入m个数表示叶结点权值。需要用这些叶结点生成哈夫曼树,输出该哈夫曼树的带权路径长度。
输入:
3
7
9 11 5 7 8 2 3
10
8 8 10 15 20 9 11 13 10 17
15
30 25 23 21 24 26 28 31 33 35 36 38 37 34 32输出:
120
398
1774
观察上面的哈弗曼树,我们可以知道
由定义知:哈夫曼树的带权路径长度等于(每个叶子结点*对应路径)的和
带权路径长度=9*2+11*2+5*3+7*3+8*3+2*4+3*4=120
而通过观察,我们又可以发现,哈夫曼树的带权路径长度=除根节点外每个结点的权值之和
带权路径长度=19+26+9+10+11+15+5+5+7+8+2+3=120
(举例的式子中,结点通过树从上往下选取)
代码:
#include<bits/stdc++.h>
using namespace std;
int main()
{
int n;
cin>>n;
while(n--)
{
int m;
cin>>m;
int a[1005];
for(int i=0;i<m;i++)
{
cin>>a[i];
}
int sum=0;
for(int i=0;i<m-1;i++)
{
sort(a+i,a+m);
sum+=a[i]+a[i+1];
a[i+1]+=a[i];
}
cout<<sum<<endl;
}
}
不用再通过构建一棵树计算,可以通过排序简单地得到我们想要的答案。