AcWing 3531. 哈夫曼树（哈夫曼树）

最新推荐文章于 2024-08-22 21:24:01 发布

5pace

最新推荐文章于 2024-08-22 21:24:01 发布

阅读量127

点赞数

分类专栏：哈夫曼树

本文链接：https://blog.csdn.net/m0_51448653/article/details/125793957

版权

优先队列小顶堆累加求和算法数据结构

关键词由CSDN通过智能技术生成

哈夫曼树专栏收录该内容

1 篇文章 0 订阅

订阅专栏

时/空限制：1s / 64MB
在这里插入图片描述
题解：

注意是pq数量为1就停下，不是为0
每次取两个，然后放入和
greater表示小顶堆

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

int main() {
    int n;
    cin >> n;
    priority_queue<int, vector<int>, greater<int>> pq;
    while (n -- ) {
        int x;
        cin >> x;
        pq.push(x);
    }
    int ans = 0;
    while (pq.size() > 1) {
        int t1 = pq.top(); pq.pop();
        int t2 = pq.top(); pq.pop();
        int now = t1 + t2;
        ans += now;
        pq.push(now);
    }
    cout << ans;
}