标签:empty arp 补充 while style 编码 输出 数据规模 表示
[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]荷马史诗
标签:empty arp 补充 while style 编码 输出 数据规模 表示
原文地址:http://www.cnblogs.com/xiao-ju-ruo-xjr/p/7073716.html