luogu P2985 [USACO10FEB]吃巧克力Chocolate Eating

题目描述

Bessie拿到了N (1 <= N <= 50,000)块巧克力。她决定想个办法吃掉这些巧克力，使得它在吃巧克力的这段时间里，最不开心的一天尽可能的开心。并且一共吃D (1 <= D <= 50,000)天。

每块巧克力有一个开心值H_i (1 <= H_i <= 1,000,000)，当某天你吃下那块巧克力时，你将获得那块巧克力的开心值。每一天的开心值是所有当天吃掉的巧克力的总开心值之和。每天晚上Bessie睡觉之后，它的开心值会减半。也就是说，比如昨天Bessie的开心值为50，那么今天早上我一醒来我就会有25点的开心值，舍去小数点后数字。另外，Bessie还有一个怪癖，她喜欢按照巧克力本来的排列顺序吃。

Bessie第一天的开心值为0，求一个每天吃巧克力的方案，使得Bessie最不开心的一天尽可能的开心。

输入格式

• Line 1: Two space separated integers: N and D

• Lines 2..N+1: Line i+1 contains a single integer: H_i

输出格式

• Line 1: A single integer, the highest Bessie's minimum happiness can be over the next D days

• Lines 2..N+1: Line i+1 contains an integer that is the day on which Bessie eats chocolate i

---

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int mod=10007,N=5e4+10;
#define int long long
int n,m,h[N],a[N];
inline bool check(int x){
    int op=0,j=0;
    for(int i=1;i<=m;i++){
        op=op>>1;
        while(op<x&&j<n){j++;op+=h[j];a[j]=i;}
        if(op<x)return 0;
    }
    return 1;
}
signed main(){
    cin>>n>>m;
    int l=0,r=10,ans=-1;
    for(int i=1;i<=n;i++)
    scanf("%lld",&h[i]),r+=h[i];
    while(l<=r){
        int mid=(l+r)>>1;
        if(check(mid)){
            l=mid+1;
            ans=mid;
        }else r=mid-1;
    }
    cout<<ans<<endl;
    memset(a,0,sizeof(a));
    check(ans);
    for(int i=1;i<=n;i++)
    if(a[i])printf("%lld\n",a[i]);
    else printf("%lld\n",m);
}