「NOI2015」荷马史诗-Huffman树

Decription

给定$n$个单词，现要求把每个词用$k$进制字符串进行替换。

每个单词有出现次数，总长度为出现次数$\times$替换字符串的长度。

求最小总长度及在保证总长度的情况下，最长字符串$s_i$的最短长度。

Solution

裸的$k$叉哈弗曼树。

将$n$个单词的加入按出现次数加入小根堆，每次取出前$k$个元素，统计他们的出现次数和$sum$,$ans+=sum$,然后将$sum$加入小根堆即可。

记得补一些权值为$0$的元素使得$(n-1) \mod (k-1)=0$

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

const int maxn = 100005;

inline long long gi()
{
    char c = getchar();
    while (c < '0' || c > '9') c = getchar();
    long long sum = 0;
    while ('0' <= c && c <= '9') sum = sum * 10 + c - 48, c = getchar();
    return sum;
}

struct node {
    long long val, dep;

    bool operator < (const node &a) const {
        return val != a.val ? val > a.val : dep > a.dep;
    }
};
priority_queue<node> que;
int n, k;
long long ans;

int main()
{
    n = gi(); k = gi();
    for (int i = 1; i <= n; ++i) que.push((node) {gi(), 0});

    while ((n - 1) % (k - 1)) que.push((node) {0, 0}), ++n;

    for (int i = 1; i <= (n - 1) / (k - 1); ++i) {
        long long sum = 0, mxdp = 0;
        for (int j = 1; j <= k; ++j) {
            sum += que.top().val; mxdp = max(mxdp, que.top().dep);
            que.pop();
        }
        ans += sum;
        que.push((node){sum, mxdp + 1});
    }

    printf("%lld\n%lld", ans, que.top().dep);

    return 0;
}