[bzoj4198]荷马史诗

很容易想到哈夫曼树，容易扩展得到贪心：优先选最小的k个合并，堆维护即可
为了保证深度最小，可以加入深度权值，在权值相同时选择当前深度最小的即可
（注意：由于可能最后一次不足k个，那么显然可以将这些不足的放到最深的位置来减小答案，需要补上权值为0的点来保证每一次都有k个）

#include<bits/stdc++.h>
using namespace std;
#define ll long long
struct ji{
    int d;
    ll val;
    bool operator < (const ji &k)const{
        return (val>k.val)||(val==k.val)&&(d>k.d);
    }
};
priority_queue<ji>q;
int n,k,x;
ll s,ans;
int main(){
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++){
        scanf("%lld",&s);
        q.push(ji{0,s});
    }
    while ((n-1)%(k-1)){
        n++;
        q.push(ji{0,0});
    }
    while (n>1){
        x=s=0;
        for(int i=1;i<=k;i++){
            s+=q.top().val;
            x=max(x,q.top().d);
            q.pop();
        }
        ans+=s;
        q.push(ji{x+1,s});
        n-=k-1;
    }
    printf("%lld
%d",ans,q.top().d);
}