哈夫曼树的本质就是在k进制下找非前缀总长最小编码,非前缀使得解码唯一,总长最小使储存和传输都更优。
对每个信息节点都有一个出现次数作为权值,权值乘上长度就是它对编码总长的贡献,问题就是要使得所有权值乘上长度之和最小。
每次取权值最小的k个点合并,把新点作为原点的父亲,新点权值为k个点的权值之和,然后剩下的n-k+1的节点再重复这个过程,(实质上是在分配k进制的0~k-1),最后节点到根的带权路径长度就是编码长度,这就是哈夫曼树的思想。
更详尽的描述与图片参见dashuai的博客,原理讲的很好。
对于k进制需要添加空节点使得最后一次合并是满的k个点。
对于建树用优先队列实现复杂度为
O
(
n
l
o
g
n
)
O(nlogn)
O(nlogn)。
如果只需要统计编码总长,参见下面这个做法,复杂度为
O
(
n
k
)
O(\frac nk)
O(kn)
相当于用前缀和和单调队列使得每次合并的复杂度为均摊O(1)
贴一个求2~n进制的编码总长异或和的代码,复杂度是 O ( n ln n ) O(n\ln n) O(nlnn)的:
#include<bits/stdc++.h>
#define maxn 2000005
#define LL long long
using namespace std;
int n,a[maxn];
LL s[maxn],q[maxn];
LL solve(int k){
int l=n+1-(k-1-(n-1)%(k-1))%(k-1),h=1,t=0;
LL ret=0,sum;
while(1){
int r=l+k-1; sum=0;
while(h<=t&&(r>2*n||(r>=l&&a[r]>q[h]))) sum+=q[h++],r--;
if(r>2*n) break;
sum+=s[r]-s[l-1];
q[++t]=sum,ret+=sum,l=r+1;
}
return ret;
}
int main()
{
//freopen("epic.in","r",stdin);
//freopen("epic.out","w",stdout);
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
sort(a+1,a+1+2*n);
for(int i=n+1;i<=2*n;i++) s[i]=s[i-1]+a[i];
LL ans=0;
for(int i=2;i<=n;i++) ans^=solve(i);
cout<<ans;
}