UVa 10954 (Huffman 优先队列) Add All

直接用一个优先队列去模拟Huffman树的建立过程。

每次取优先队列前两个数，然后累加其和，把这个和在放入到优先队列中去。

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

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

    return 0;
}