树-堆结构练习——合并果子之哈夫曼树

A - 树-堆结构练习——合并果子之哈夫曼树

Time Limit:1000MS     Memory Limit:65536KB     64bit IO Format:%lld & %llu

Submit   Status

Description

 在一个果园里，多多已经将所有的果子打了下来，而且按果子的不同种类分成了不同的堆。多多决定把所有的果子合成一堆。

每一次合并，多多可以把两堆果子合并到一起，消耗的体力等于两堆果子的重量之和。可以看出，所有的果子经过n-1次合并之后，就只剩下一堆了。多多在合并果子时总共消耗的体力等于每次合并所消耗体力之和。

Input

 第一行是一个整数n(1<=n<=10000),表示果子的种类数。第二行包含n个整数，用空格分隔，第i个ai(1<=ai<=20000)是第i个果子的数目。

 

Output

 输出包括一行，这一行只包含一个整数，也就是最小的体力耗费值。输入数据保证这个值小于2^31。

 

Sample Input

3
1 2 9

Sample Output

15

Hint

题意很好理解，用的c++当中的优先队列，刚开始不会，但是看了正南lee的写的很详细就明白了

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>

#include <queue>//队列的头文件

using namespace std;

int a[10000];

int main()

{
    int n,a1,a2,i;

    int sum=0;
    cin>>n;
    for(i=0;i<n;i++)
    {
        cin>>a[i];
    }
    priority_queue<int , vector<int> , greater<int> >q;//从小到大排列
    for(i=0;i<n;i++)
    {
        q.push(a[i]);//进入队列
    }
    while(q.size()>1)//因为最后只剩下一组，所以控制长度大于1
    {
        a1=q.top();//将a1定义为首元素
        q.pop();//出队列
        a2=q.top();//将a2定义为首元素
        q.pop();//出队列
        sum+=a1+a2;
        q.push(a1+a2);//这里注意将a1+a2进入队列，而不是sum
    }
    cout<<sum<<endl;//最后输出sum
    return 0;
}