问题描述
给定一段文字,已知单词a1, a2, …, an出现的频率分别t1, t2, …, tn。可以用01串给这些单词编码,即将每个单词与一个01串对应,使得任何一个单词的编码(对应的01串)不是另一个单词编码的前缀,这种编码称为前缀码。
使用前缀码编码一段文字是指将这段文字中的每个单词依次对应到其编码。一段文字经过前缀编码后的长度为:
L=a1的编码长度×t1+a2的编码长度×t2+…+ an的编码长度×tn。
定义一个前缀编码为字典序编码,指对于1 ≤ i < n,ai的编码(对应的01串)的字典序在ai+1编码之前,即a1, a2, …, an的编码是按字典序升序排列的。
例如,文字E A E C D E B C C E C B D B E中, 5个单词A、B、C、D、E出现的频率分别为1, 3, 4, 2, 5,则一种可行的编码方案是A:000, B:001, C:01, D:10, E:11,对应的编码后的01串为1100011011011001010111010011000111,对应的长度L为3×1+3×3+2×4+2×2+2×5=34。
在这个例子中,如果使用哈夫曼(Huffman)编码,对应的编码方案是A:000, B:01, C:10, D:001, E:11,虽然最终文字编码后的总长度只有33,但是这个编码不满足字典序编码的性质,比如C的编码的字典序不在D的编码之前。
在这个例子中,有些人可能会想的另一个字典序编码是A:000, B:001, C:010, D:011, E:1,编码后的文字长度为35。
请找出一个字典序编码,使得文字经过编码后的长度L最小。在输出时,你只需要输出最小的长度L,而不需要输出具体的方案。在上面的例子中,最小的长度L为34。
输入格式
输入的第一行包含一个整数n,表示单词的数量。
第二行包含n个整数,用空格分隔,分别表示a1, a2, …, an出现的频率,即t1, t2, …, tn。请注意a1, a2, …, an具体是什么单词并不影响本题的解,所以没有输入a1, a2, …, an。
输出格式
输出一个整数,表示文字经过编码后的长度L的最小值。
样例输入
5
1 3 4 2 5
样例输出
34
样例说明
这个样例就是问题描述中的例子。如果你得到了35,说明你算得有问题,请自行检查自己的算法而不要怀疑是样例输出写错了。
评测用例规模与约定
对于30%的评测用例,1 ≤ n ≤ 10,1 ≤ ti ≤ 20;
对于60%的评测用例,1 ≤ n ≤ 100,1 ≤ ti ≤ 100;
对于100%的评测用例,1 ≤ n ≤ 1000,1 ≤ ti ≤ 10000。
解题说明
题目中说到哈夫曼编码,它与题目要求的数据压缩方式就差一个字典序。哈夫曼编码就是不管什么abcd的顺序,每次都把最小的子树合并起来,而这里的压缩编码方式就要求先abcd…排成一条线,然后只允许相邻的两个子树进行合并。比如说我随便弄两种满足字典序的编码:
- 左边:
字符 | 编码 |
---|---|
A | 0000 |
B | 0001 |
C | 001 |
D | 01 |
E | 1 |
总长 | 1x4+3x4+4x3+2x2+5x1=37 |
- 右边:
字符 | 编码 |
---|---|
A | 0 |
B | 100 |
C | 101 |
D | 110 |
E | 111 |
总长 | 1x1+3x3+4x3+2x3+5x3=43 |
至少可以看出编码后字符的长度等于”每个字符出现的频数x字符在二叉树中的层数“之和,而层数换种说法就是该字符被合并的次数。如果之前了解”石子合并“的问题,想到这就应该明白这个压缩编码的问题其实就等同于石子合并,石子合并的结果就是”每个石头堆的石子数x被合并的次数“之和。
字 符 ----> 石子堆
频 数 ----> 石头数量
子树合并 ----> 石子堆合并
编码长度 ----> 游戏得分
ps:石子合并的问题网上有很多很好的文章,其实我也是看别人的’>_<’。
#include <iostream>
#include <cstdio>
#define N 1005
#define INF 0x7fffffff
using namespace std;
int dp[N][N];
int sum[N];
int a[N];
int run(int a[],int n){
for(int i=0;i<n;i++){
dp[i][i]=0;
}
for(int v=1;v<n;v++){
for(int i=0;i<n-v;i++){
int j=i+v;
dp[i][j]=INF;
int s=sum[j]-sum[i]+a[i];
for(int k=i;k<j;k++){
dp[i][j]=min(dp[i][k]+dp[k+1][j]+s,dp[i][j]);
}
}
}
return dp[0][n-1];
}
int main(void){
int n;
cin>>n;
for(int i=0;i<n;i++){
cin>>a[i];
}
sum[0]=a[0];
for(int i=1;i<n;i++){
sum[i]=sum[i-1]+a[i];
}
cout<<run(a,n)<<endl;
return 0;
}