首页 > itarticle > uva 10954 – add all

uva 10954 – add all

admin 11月 12, 2020 0

Contents

Problem

Solution

利用 Huffman Code 的方法，將較小的數字先行結合成另一個數字，一邊計算其代價。

Code


#include<queue>
#include<vector>
#include<functional>
using namespace std;

int ()
{
    int n;
    priority_queue<int, vector<int>, greater<int> > PQ;

    while (scanf("%d", &n) && n)
    {
        int num;
        long long sum = 0;
        for (int i = 0; i < n; i++)
        {
            scanf("%d", &num);
            PQ.push(num);
        }

        for (int i = 1; i < n; i++)
        {
            int a, b;
            a = PQ.top(), PQ.pop();
            b = PQ.top(), PQ.pop();

            sum += (a + b);
            PQ.push(a + b);
        }

        PQ.pop();
        printf("%lldn", sum);
    }

    return 0;
}