哈夫曼树,第一行输入一个数n,表示叶结点的个数。
需要用这些叶结点生成哈夫曼树,根据哈夫曼树的概念,这些结点有权值,即weight,题目需要输出哈夫曼树的带权路径长度(WPL)。
输入格式:
第一行输入一个数n,第二行输入n个叶结点(叶结点权值不超过1000,2<=n<=1000)。
输出格式:
在一行中输出WPL值。
输入样例:
5
1 2 2 5 9
结尾无空行
输出样例:
37
#include<bits/stdc++.h>
using namespace std;
int main()
{
priority_queue<int,vector<int>,greater<int> >q;//优先队列,递增
int n,x,sum=0;
cin>>n;
for(int i=0;i<n;i++)
{
cin>>x;
q.push(x);
}
while(q.size()>1)
{
int top=q.top();
q.pop();
int k=q.top();
k+=top;
q.pop();
sum+=k;
q.push(k);
}
cout<<sum;
return 0;
}