Huffman编码-荷马史诗 NOI2015 Day2

题目被吃了；Huffman编码的具体内容网上资料很多，所以被吃了。

建 Huffman 树方式如下：

将每个关键词的权值（这道题里就是这个词出现的次数）作为第一关键字，

其深度作为第二关键词，放入大根堆中，取堆顶前 k 个元素作为一个新的

节点的子节点。最终能合并为一棵树的条件是  (n-1) mod (k-1) = 0。

如果不满足，应该往堆里扔适量权值为 0 的“假节点”。

注意：

1.对于这道题，如果只将权值作第一关键字，只能得 55 分。一定要将深度作为第二关键词建堆！

2.关于最后文章的总长度，应该是 每个节点的深度 之和，但是并不需要枚举节点进行计算。

根据树上DP的思想，每次合并时将深度累加即可，代码中用 sum 实现了这一点。

 

#include <stdio.h>
#include <queue>

using namespace std;

typedef long long LL;

struct node {
    LL v, deep;
    node (LL v, LL deep):
        v(v), deep(deep) { }
};

bool operator < (const node& t1, const node& t2) { return t1.v > t2.v || t1.v == t2.v && t1.deep > t2.deep; }

priority_queue<node> Q;

LL n, k, ans;

int main()
{
    LL i, tmp;
    scanf("%lld%lld", &n, &k);
    if ((tmp = (n-1)%(k-1))) {
        tmp = k-1-tmp;
        while (tmp--) Q.push(node(0, 0));
    }
    for (i = 1; i <= n; ++i)
        scanf("%lld", &tmp), Q.push(node(tmp, 0));
    ans = 0;
    while (Q.size() != 1) {
        LL sum = 0, maxx = 0;
        for (i = 1; i <= k; ++i)
            sum += Q.top().v, maxx = max(maxx, Q.top().deep), Q.pop();
        Q.push(node(sum, maxx+1));
        ans += sum;
    }
    printf("%lld\n%lld\n", ans, Q.top().deep);
    return 0;
}