BZOJ4198(Noi2015)[荷马史诗]--k叉哈夫曼树

【链接】
bzoj4198

【题目大意】

求n个字符串，满足这个字符串是k进制字符串，每个字符只包括0~k-1中的整数，且满足对于$1<=i,j<=n$,si不是sj的前缀。求如何使n个不同字符串的长度乘相应个数的总长度最小，且最长的si的最短长度。

【解题报告】

第一问就是裸的哈弗曼编码。

所以这个问题就变成了求一颗k叉哈夫曼树的点权总和的最小值，其实题目中给定的n可能并不能构造一颗完全k叉哈夫曼树，所以我们可以往其中添加一些权值为0的节点，是这棵树是完全k叉哈夫曼树。

对于求第二问，其实就是求这个k叉哈夫曼树的最大深度的最小值。所以在开个数组标记一下深度就出解了。

#include<cstdio>
#include<algorithm>
#define LL long long
using namespace std;
const int maxn=100015;
int n,m,len,tot;
LL ans;
struct Node
{
    LL x; int d;
    bool operator < (const Node a) const{
        return x<a.x||(x==a.x&&d<a.d);
    }
}hep[maxn];
inline LL Read()
{
    LL res=0;
    char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9') res=res*10+ch-48,ch=getchar();
    return res;
}
void Put(LL x,int d)
{
    hep[++len]=(Node){x,d}; int son=len;
    while (son!=1&&hep[son]<hep[son>>1]) swap(hep[son],hep[son>>1]),son>>=1;
}
Node Get()
{
    Node sum=hep[1]; hep[1]=hep[len--]; int fa=1,son;
    while (fa*2<=len)
    {
        if (fa*2+1>len||hep[fa*2]<hep[fa*2+1]) son=fa*2; else son=fa*2+1;
        if (hep[son]<hep[fa]) swap(hep[fa],hep[son]),fa=son; else break;
    }
    return sum;
}
int max(int x,int y) {if (x>y) return x; return y;}
int main()
{
    freopen("4198.in","r",stdin);
    freopen("4198.out","w",stdout);
    n=Read(); m=Read(); len=tot=ans=0;
    for (int i=1; i<=n; i++) Put(Read(),0);
    if (m>2) while (n%(m-1)!=1) n++,Put(0,0);
    while (len>=m)
    {
        int MAX=0; LL sum=0;
        for (int i=1; i<=m; i++) {Node x=Get(); sum+=x.x; MAX=max(MAX,x.d);}
        MAX++; Put(sum,MAX); tot=max(tot,MAX); ans+=sum;
    }
    printf("%lld\n%d\n",ans,tot);
    return 0;
}