LOJ#2132. 「NOI2015」荷马史诗

$n leq 100000$个数字，放进$k$叉树里，一个点只能放一个数，使所有数字乘以各自深度这个值之和最小的同时，最大深度的数字最小。

哈夫曼。这是我刚学OI那段时间看到的，感觉就是个很无聊的贪心，而且密码学我也不学深对哈夫曼的应用也了解不多，没想到出现在noi。

原来的哈夫曼只需要每次拿k个最小的数出来，建一个他们共同的父亲并在一起，当作一个权值为他们权值之和的新点，用堆可以实现；由于$(n-1) mod (k-1)$不一定为0，需要补几个0点进去。相对于原来的哈夫曼，这里多了个深度限制，那只需要把堆里元素再记一下最大深度就可以了。

//#include<iostream>
#include<cstring>
#include<cstdio>
//#include<math.h>
//#include<set>
#include<queue>
//#include<bitset>
//#include<vector>
#include<algorithm>
#include<stdlib.h>
using namespace std;

#define LL long long
int qread()
{
    char c; int s=0,f=1; while ((c=getchar())<'0' || c>'9') (c=='-') && (f=-1);
    do s=s*10+c-'0'; while ((c=getchar())>='0' && c<='9'); return s*f;
}

//Pay attention to '-' , LL and double of qread!!!!

int n,K;
#define maxn 200011
struct qnode
{
    LL v; int dep;
    bool operator > (const qnode &b) const {return v>b.v || (v==b.v && dep>b.dep);}
};
priority_queue<qnode,vector<qnode>,greater<qnode> > q;

int main()
{
    n=qread(); K=qread(); LL v;
    for (int i=1;i<=n;i++) {scanf("%lld",&v); q.push((qnode){v,0});}
    if ((n-1)%(K-1)) for (int i=1,to=(K-1)-(n-1)%(K-1);i<=to;i++) q.push((qnode){0,0}),n++;
    LL ans=0;
    for (int i=1,to=(n-1)/(K-1);i<=to;i++)
    {
        LL nv=0; int nd=0;
        for (int j=1;j<=K;j++) nv+=q.top().v,nd=max(nd,q.top().dep),q.pop();
        ans+=nv; q.push((qnode){nv,nd+1});
    }
    printf("%lld
%d
",ans,q.top().dep);
    return 0;
}