哈夫曼—谈判

题目描述

在很久很久以前，有 nn个部落居住在平原上，依次编号为 1 到n。第 i 个部落的人数为 ti​。

有一年发生了灾荒。年轻的政治家小蓝想要说服所有部落一同应对灾荒，他能通过谈判来说服部落进行联合。

每次谈判，小蓝只能邀请两个部落参加，花费的金币数量为两个部落的人数之和，谈判的效果是两个部落联合成一个部落（人数为原来两个部落的人数之和）。

输入描述

输入的第一行包含一个整数 n，表示部落的数量。

第二行包含 n个正整数，依次表示每个部落的人数。

其中，1≤n≤1000，1≤ti​≤10^4。

输出描述

输出一个整数，表示最小花费。

输入输出样例

示例 1

输入

4
9 1 3 5

输出

31

运行限制

• 最大运行时间：1s
• 最大运行内存: 128M

解题思路：

数据范围很小的情况下可以考虑暴力求解，但是这里我们使用的是优先队列，自动排序，但是注意C++中的优先队列是一种大顶堆，所以存放数据的时候可以存放相反数，但是取出数据进行求和的时候再使用相反数求得原数，就可以使得复杂度较低。

代码：

#include <bits/stdc++.h>
using namespace std;
priority_queue<int> a;//优先队列是一个大顶堆，按照从大到小排序的
int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        int x;
        cin>>x;
        a.push(-x);
    }
    int sum=0;
    if(a.size()==1)
    {
        sum=-a.top();
    }
    else
    {
        while(a.size()>1)
        {
            int x=a.top();
            a.pop();
            int y=a.top();
            a.pop();
            a.push(x+y);
            sum+=(-x-y);
        }
    }
    cout<<sum;
    return 0;
}