天天看點

合并果子之哈夫曼樹(優先隊列實作)

#include<bits/stdc++.h>

using namespace std;

/* 優先隊列用法總結*/
/*priority_queue<Type, Container, (Functional)> Q;
Type      : 隊列類型
Container : 給優先隊列一個訓示 内部用什麼容器實作 預設vector
Functional: less<Type> 類似于大頂堆 将大的數放于前面
                priority_queue<Type, Container, less<int>> Q1;
                priority_queue<Type, Container> Q2;
                Q1 = Q2;
            greater<Type>  将小的數放于隊首
            equal_to<Type> 等于(我也不知道怎麼用2333)
*/
int main()
{
    int n;
    while(cin >> n)
    {
        priority_queue<int, vector<int>, greater<int> > Q;
        for(int i = 0; i < n; i++)
        {
            int x;
            cin >> x;
            Q.push(x);
        }
        int cnt = 0;
        while(!Q.empty())
        {
            int a = Q.top();
            Q.pop();
            if(!Q.empty()) /// 表示内部至少還有兩個元素a,b 将其合并
            {
                int b = Q.top();
                Q.pop();
                Q.push(a+b);
                cnt = cnt + b + a;
            }
        }
        cout << cnt << endl;
    }
    return 0;
}

           

繼續閱讀