4198: [Noi2015]荷马史诗 贪心+优先队列

其实我并不太懂Huffman编码的原理什么的。。只是套K叉哈弗曼树罢了。。
然而由于初赛一直随便水所以哈弗曼树是什么我也不太懂只知道是贪心。。

#include<bits/stdc++.h>
#define ll long long 
using namespace std;
int n,k;
ll ans;
priority_queue<pair<ll,int>,vector<pair<ll,int> >,greater<pair<ll,int> > > q;
inline ll read()
{
    ll a=0,f=1; char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
    while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
    return a*f;
}
int main()
{
    n=read(); k=read();
    for (int i=1;i<=n;i++) q.push(make_pair(read(),0));
    while (k!=2&&(n-1)%(k-1)) n++,q.push(make_pair(0,0));
    while (q.size()>1)
    {
        ll x=0; int y=0;
        for (int i=1;i<=k;i++)
            x+=q.top().first,y=max(y,q.top().second),q.pop();
        ans+=x; q.push(make_pair(x,y+1));
    }
    cout << ans << endl << q.top().second << endl;
    return 0;
}