优先队列和哈夫曼树

Posted 2021-03-22 clarencezzh

　　https://www.acwing.com/problem/content/description/150/

 

哈夫曼树可以通过小根堆实现。小根堆每次弹出两个值，然后将二者的和再插入小根堆中。

 

　　比如求

 

#include<stdio.h>
#include<queue>
using namespace std;
int n,ans,k;
priority_queue<int,vector<int>,greater<int> > q;//小根堆
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&k);
        q.push(k);
    }
    for(int i=1;i<n;i++){
        int a=q.top();
        q.pop();
        int b=q.top();
        q.pop();
        ans+=a+b;
        q.push(a+b);
    }
    printf("%d",ans);
    return 0;
}