题解 | 哈夫曼树

#include <bits/stdc++.h>
using namespace std;


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 ans=0;
        while(q.size()>1){
            int x=q.top();
            q.pop();
            int y=q.top();
            q.pop();
            ans+=x+y;
            q.push(x+y);
        }
        cout<<ans<<endl;
    }
}

我们都知道，二叉哈夫曼树的生成方法就是取两个最小的元素加在一起，放在最下面，那么就利用这个特性，采用优先队列的greater形式，把最小的放在最上面，那么就可以得到最小的两个元素，这时候，我们取出来，加到ans里面，注意，这时候，我们会把最小的两个元素的和放回原来的优先队列，这时候就可以完美得到ans*边数，就得到WSL