【洛谷P2168】[NOI2015]荷马史诗

荷马史诗

建一个k叉哈夫曼树，用堆维护一下

// luogu-judger-enable-o2
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
#define MAXN 1000010
#define max(i,j) ((i)>(j)?(i):(j))
#define int long long
int n,ans,k;
struct NODE{
    int w,h;
} a[MAXN];
struct cmp{
    bool operator()(NODE x,NODE y){　
        if(x.w!=y.w) return x.w>y.w;　 // 先按代价大小排序
        else return x.h>y.h;　　　　　　// 再按子树深度排序
    }
};
inline int read(){
    int x=0; char c=getchar();
    while(c<'0'||c>'9') c=getchar();
    while('0'<=c&&c<='9') { x=(x<<3)+(x<<1)+c-'0'; c=getchar(); }
    return x;
}
priority_queue< NODE, vector < NODE > , cmp > q0,q;
#undef int
int main()
#define int long long
{
    n=read(); k=read();
    int x; NODE p0;
    p0.w=0; p0.h=1;
    for(int i=1;i<=n;i++){
        x=read(); NODE p;
        p.w=x; p.h=1; q0.push(p);
    }
    q=q0; ans=0;
    int top=0;
    if((n-1)%(k-1)!=0) top=k-1-(n-1)%(k-1);
    for(int i=1;i<=top;i++) q.push(p0);
    while(q.size()>1){
        NODE s=p0;
        for(int i=1;i<=k;i++){
            NODE b=q.top(); q.pop();
            ans+=b.w; s.w+=b.w; s.h=max(s.h,b.h);
        }
        s.h++;
        q.push(s);
    }
    NODE t=q.top();
    printf("%lld
%lld",ans,t.h-1);
    return 0;
}