題目連結

題目大意：一篇文章有n個單詞，其中第i個單詞的出現次數為w[i]。你要用k進位制串s[i]替換第i種單詞。要求：對於任意i!=j，都有s[i]不是s[j]的字首。
要使替換後的文章總長度最小。
求這個最小總長度。
在這個前提下，要使s[i]的最大長度最短，求這個最短的長度

題解：k叉哈夫曼樹
取最小的k個合併即可
發現只有當(n-1)%(k-1)=0的時候才能恰好合並，所以要新增k-1-(n-1)%(k-1)個權值為0高度為1的虛擬節點
用堆維護就好了

為了保證最大高度最小，把高度加成第二關鍵字

我的收穫：姿勢++

#include <bits/stdc++.h>
 

using namespace std; 

#define ll long long

ll n,k,ans;

struct node{
    ll val,dep;
    node(){}
    node(ll _,ll __){val=_,dep=__;}
}tmp;

bool operator <(node x,node y){return x.val>y.val||(x.val==y.val&&x.dep>y.dep);}

priority_queue<node> q;

ll readin()
{
    ll x=0;char
 
 ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch>='0'&&ch<='9') x=x*10+(ll)(ch-'0'),ch=getchar();
    return x;
}

void work()
{
    while(n>1){
        ll maxd=0,tot=0;
        for(ll i=1;i<=k;i++){
            tmp=q.top();q.pop();
            tot+=tmp.val;
            maxd=max(maxd,tmp.dep);
        }  
        q.push(node(tot,maxd+1
 
));
        ans+=tot;n-=k-1;
    }
    printf("%lld\n%lld\n",ans,q.top().dep-1);
}

void init()
{
    n=readin();k=readin();
    for(ll i=1;i<=n;i++) q.push(node(readin(),1));
    ll remain=(n-1)%(k-1);
    if(remain!=0) remain=k-1-remain,n+=remain;
    for(ll i=1;i<=remain;i++) q.push(node(0,1));
}

int main()
{
    init();
    work();
    return 0;
}