[UOJ#130][BZOJ4198][Noi2015]荷马史诗
试题描述
追逐影子的人，自己就是影子。 ——荷马
输入
输入文件的第 1 行包含 2 个正整数 n,k，中间用单个空格隔开，表示共有 n 种单词，需要使用 k 进制字符串进行替换。
输出
输出文件包括 2 行。
输入示例
6 3 1 1 3 3 9 9
输出示例
36 3
数据规模及约定
对于所有数据，保证 2≤n≤100000，2≤k≤9。
题解
这题没啥好说的，k 进制 Huffman 编码。
注意当 (n-1) mod (k-1) > 0 时，需要补 0，因为最终的 Huffman 树（其实就是 Trie 树）必须是一个完全 k 叉树。
题目还要求最大深度最小，那么堆中比较大小的规则补充一下：当权重相同时，优先取深度更小的子树。
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cctype>
#include <algorithm>
#include <queue>
using namespace std;
#define LL long long
const int BufferSize = 1 << 16;
char buffer[BufferSize], *Head, *Tail;
inline char Getchar() {
if(Head == Tail) {
int l = fread(buffer, 1, BufferSize, stdin);
Tail = (Head = buffer) + l;
}
return *Head++;
}
LL read() {
LL x = 0, f = 1; char c = Getchar();
while(!isdigit(c)){ if(c == ‘-‘) f = -1; c = Getchar(); }
while(isdigit(c)){ x = x * 10 + c - ‘0‘; c = Getchar(); }
return x * f;
}
#define maxn 100010
struct Node {
LL val; int mxd;
Node() {}
Node(LL _, int __): val(_), mxd(__) {}
bool operator < (const Node& t) const { return val != t.val ? val > t.val : mxd > t.mxd; }
};
priority_queue <Node> Heap;
int n, K;
int main() {
n = read(); K = read();
for(int i = 1; i <= n; i++) Heap.push(Node(read(), 0));
while((n - 1) % (K - 1)) Heap.push(Node(0, 0)), n++;
LL ans = 0;
while(Heap.size() > 1) {
LL sumv = 0; int mxd = 0;
for(int c = 1; c <= K && !Heap.empty(); c++) {
Node u = Heap.top(); Heap.pop();
sumv += u.val; mxd = max(mxd, u.mxd);
}
ans += sumv; mxd++;
Heap.push(Node(sumv, mxd));
}
printf("%lld\n%d\n", ans, Heap.top().mxd);
return 0;
}
[UOJ#130][BZOJ4198][Noi2015]荷马史诗
原文：http://www.cnblogs.com/xiao-ju-ruo-xjr/p/7073716.html