【bzoj4198】[NOI2015]荷马史诗

= =怪不得说15年的noi水，我这个傻逼都会做，就是一个k-哈夫曼树，小模型可以去看noip的合并果子，直接贪心搞就好了，用两个数组，跟切蚯蚓一样比较，不过它是三个数组，然后选最小k个合并，需要注意的是当（n-1）%(k-1)!=0的时候，需要先合并一次，以防最后一次合并的时候不足k个，至于第二问，合并的时候定义一个结构体递推就好了，附代码

#include<iostream>#include<cstdio>#include<cstring>#include<string>#include<algorithm>#define max(a,b) (a>b?a:b)using namespace std;const int N=100005;int  ak,k,n,cnt,cura,curb,jib;long long ans,pd,a[N];struct data{    long long q;    int c;}b[N];int judge(){    if (cura>n||(a[cura]>b[curb].q&&curb<=jib)){pd+=b[curb++].q;return b[curb-1].c;}    else {pd+=a[cura++];return 0;}}int main(){    cin>>n>>k;    for (int i=1;i<=n;i++)    scanf("%lld",&a[i]);cura=curb=1;jib=0;    int pk=(n-1) % (k-1),ci=(n-1)/(k-1);    sort(a+1,a+n+1);cnt=ans=0;    if (pk>0)    {        for (int i=1;i<=pk+1;i++)        ans+=a[i];        b[++jib].c=1,b[jib].q=ans;cura+=pk+1;    }    for (int i=1;i<=ci;i++)    {        cnt=pd=0;        for (int j=1;j<=k;j++)        ak=judge(),cnt=max(cnt,ak);        b[++jib].c=cnt+1;b[jib].q=pd;        ans+=pd;    }    cout<<ans<<endl<<b[jib].c;    return 0;}