数据结构综合应用——堆

小复习：哈夫曼树的构造方法，每次找到权值最大的两个节点，合并，删掉，加入树，即堆的典型应用，用堆实现贪心的过程

洛谷2168

题意：将单词按照出现次数重新编码，使新编排的史诗实现哈夫曼树

思路：K叉哈夫曼树
每次找到最小的k个节点，合并成一个节点，加入堆中
有可能使得根节点处不满造成根节点处浪费，要先补充几个权值为0的节点
当一个k叉树有m个中间节点，则有m(k - 1) + 1个叶子节点，所以补充n / (k + 1)余数为1个节点即可

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

struct node {
    ll w = 0, h = 0;
    node(ll w, ll h) : w(w), h(h) {} //构造函数，如第24的调用
    bool operator < (const node &a) const { //重载运算符
        return a.w == w ? h > a.h : w > a.w;
    }
};

ll ans;
priority_queue<node> q; //存放所有节点的权值

int main() {
    ll n, k;
    ans = 0;
    scanf("%lld%lld", &n, &k);
    for (int i = 1; i <= n; i++) {
        ll w;
        scanf("%lld", &w);
        q.push(node(w, 1));
    }
    while ((q.size() - 1) % (k - 1) != 0)
        q.push(node(0, 1)); //补充节点
    while (q.size() >= k) {
        ll h = -1;
        ll w = 0;
        for (int i = 1; i <= k; i++) {
            node t = q.top();
            q.pop();
            h = max(h, t.h);
            w += t.w;
        }
        ans += w;
        q.push(node(w, h + 1));
    }
    printf("%lld\n%lld\n", ans, q.top().h - 1);
    return 0;
}

“进阶算法”专栏（目录）