acwing148. 合并果子

分析

利用哈夫曼算法:用小根堆存储重量,每次选最小两个合并。

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

int n;
int main(){
    cin>>n;
    priority_queue<int ,vector<int>,greater<int>>heap;
    while(n--){
        int x;
        cin>>x;
        heap.push(x);
    }
    int res=0;
    while(heap.size()>1){
        int a=heap.top();heap.pop();
        int b=heap.top();heap.pop();
        res+=a+b;
        heap.push(a+b);
    }
    cout<<res<<endl;
    return 0;
}