题目描述
给定n个权值作为n个叶子结点,构造哈夫曼树, 求其带权路径长度。
输入
输入由多组数据组成。
每组数据分成两行。第一行仅一个整数n(2<=n<=100000)。第二行有n个空格分开的权值,值范围在[1,1000000000]之间。
输出
对于每组测试数据,输出一行,即其对应哈夫曼树的带权路径长度对1000000007取模。
样例输入
4
7 5 2 4
8
5 29 7 8 14 23 3 11
样例输出
35
271
提示
给定n个权值作为n个叶子结点,构造哈夫曼树, 求其带权路径长度。
输入
输入由多组数据组成。
每组数据分成两行。第一行仅一个整数n(2<=n<=100000)。第二行有n个空格分开的权值,值范围在[1,1000000000]之间。
输出
对于每组测试数据,输出一行,即其对应哈夫曼树的带权路径长度对1000000007取模。
样例输入
4
7 5 2 4
8
5 29 7 8 14 23 3 11
样例输出
35
271
提示
注意运算溢出
思路:看到题目首先想到了模拟哈夫曼树的构造,然后遍历计算,但数据100000,很大,构造的话很容易超时,后来网上搜索到了哈夫曼树的公式
http://blog.csdn.net/u013011841/article/details/38226099
然后用sort试了下发现,超时,后来想到了用优先队列priority_queue和multiset可以在输入的时候默认的从小到大排列,but此时自己数学学的不是很好,对每一步进行了去模运算,错了5次,最后确认是只在最后的结果取模就可以了。
#include <queue>
#include <cstdio>
#include <iostream>
#include<algorithm>
using namespace std;
#define MOD 1000000007
typedef long long ll;
int main()
{
ll n;
while(scanf("%lld",&n)==1)
{
ll m,re=0;
priority_queue<ll , vector<ll >, greater<ll> >p;
if(!p.empty())
p.pop();
for(ll i=1;i<=n;i++){
cin>>m;
p.push(m);
}
/*for(int i=1;i<=n;i++)
{
int a=p.top();
cout<<a;
p.pop();
}*/
for(ll i=2;i<=n;i++){
ll a=p.top(); p.pop();
ll b=p.top(); p.pop();
re=a+b+re;
p.push(a+b);
// cout<<a<<b<<endl;
}
cout<<re%MOD<<endl;
}
return 0;
}