重铸华农荣光 我辈义不容辞

Day 28 - huffman

这样的 Huffman 十分 brief and elegant

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

int main()
{
    int n;
    cin>>n;
    priority_queue<int,vector<int>,greater<int>> heap;

    while(n--)
    {
        int x;
        cin>>x;
        heap.push(x);
    }

    int ans=0;
    while(heap.size()>1)
    {
        int a=heap.top();heap.pop();
        int b=heap.top();heap.pop();
        ans+=a+b;
        heap.push(a+b);
    }

    cout<<ans;
    return 0;
}