【bzoj4198】[NOI2015]荷马史诗

最新推荐文章于 2018-02-20 11:06:00 发布

黑猫black

最新推荐文章于 2018-02-20 11:06:00 发布

阅读量409

点赞数

分类专栏：贪心

本文链接：https://blog.csdn.net/zhaoyh2000/article/details/54310412

版权

贪心专栏收录该内容

2 篇文章 0 订阅

订阅专栏

= =怪不得说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;
}