UVa 10954 - Add All

链接：

https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=1895

 

题意：

有n（n≤5000）个数的集合S，每次可以从S中删除两个数，然后把它们的和放回集合，直到剩下一个数。
每次操作的开销等于删除的两个数之和，求最小总开销。所有数均小于1e5。

 

分析：

用一个优先队列代表集合S，每次从中取出两个最小的数相加，再把这两个数的和放回S中即可。
与 Huffman 编码的建立过程类似。

 

代码：

#include <cstdio>
#include <queue>
using namespace std;

int main(){
    int n;
    while(scanf("%d", &n) && n){
        priority_queue<int, vector<int>, greater<int> > Q;
        for(int x, i = 0; i < n; i++){
            scanf("%d", &x);
            Q.push(x);
        }
        int ans = 0;
        while(Q.size() > 1){
            int a = Q.top();  Q.pop();
            int b = Q.top();  Q.pop();
            ans += a + b;
            Q.push(a + b);
        }
        printf("%d\n", ans);
    }
    return 0;
}