【BZOJ 4198】[Noi2015]荷马史诗 哈夫曼编码

#include <cstdio>
#include <cstring>
#include <algorithm>
#define HoN Heap::
#define make(a,b) ((Heap::V){(a),(b)})
typedef long long LL;
const int N=100010;
namespace Heap{
  struct V{
    LL val;int deep;
    inline friend bool operator <(V a,V b);
  }k[N];
  int len;
  inline bool operator <(V a,V b){
    return a.val<b.val||(a.val==b.val&&a.deep<b.deep);
  }
  inline bool empty(){return len==0;}
  inline V top(){return k[1];}
  inline int size(){return len;}
  inline void pop(){
    k[1]=k[len--];register int now=1;
    while(now<=(len>>1)){
      int next=now<<1;
      if(next<len&&k[next|1]<k[next])next|=1;
      if(k[now]<k[next])return;
      std::swap(k[now],k[next]),now=next;
    }
  }
  inline void push(V key){
    k[++len]=key;register int now=len;
    while(now!=1&&k[now]<k[now>>1])
      std::swap(k[now],k[now>>1]),now>>=1;
  }
}
int n,k;
LL ans;
int main(){
  scanf("%d%d",&n,&k);LL x;
  for(int i=1;i<=n;++i)
    scanf("%lld",&x),HoN push(make(x,1));
  if(k!=2&&n%(k-1)!=1){
    if(n%(k-1)==0)HoN push(make(0,1)),++n;
    else{
      for(int i=1;i<=k-(n%(k-1));++i)
        HoN push(make(0,1));
      n+=n%(k-1);
    }
  }
  int m=k==2?n-1:n/(k-1);
  int max;HoN V use;
  while(m--){
    x=0,max=0;
    for(int i=1;i<=k;++i)
      use=HoN top(),HoN pop(),x+=use.val,max=std::max(max,use.deep);
    ans+=x,HoN push(make(x,max+1));
  }
  printf("%lld\n%d",ans,HoN top().deep-1);
  return 0;
}