19 hdu多校 Distribution of books //线段树+dp+二分

http://acm.hdu.edu.cn/showproblem.php?pid=6606

题意：n本书，每本书有一个愉悦值ai，可以删去[x,n]的书，将剩下的书连续的分成k份，给k个人，求最小的愉悦最大值。

思路：

二分最小的愉悦值，dp的话，定义dp[i]为i结束的可以最多分成的份数，转移显然是n^2的复杂度。

先离散化前缀和，再用线段树维护以每一个离散后的值结束的最大分成的份数。

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define FI first
#define SE second
#define MP make_pair
#define PII pair<int,int>
const LL mod = 1e9+7;
const int MX = 1e6+5;
int a[MX];
int dp[MX];
vector<LL>pa;
int n,k;
void pushUp(int rt){
    dp[rt]=max(dp[rt<<1],dp[rt<<1|1]);
}
void build(int l,int r,int rt){
    if(l==r){dp[rt]=-1;return;}
    int mid=(l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    pushUp(rt);
}
void update(int l,int r,int rt,int p,int c){
    if(l==r) {dp[rt]=c;return ;}
    int mid=(l+r)>>1;
    if(p<=mid) update(l,mid,rt<<1,p,c);
    else update(mid+1,r,rt<<1|1,p,c);
    pushUp(rt);
}
int query(int L,int R,int l,int r,int rt){
    if(l>=L&&r<=R) return dp[rt];
    int mid=(l+r)>>1;
    int mx=INT_MIN;
    if(mid>=L) mx=max(mx,query(L,R,l,mid,rt<<1));
    if(mid<R) mx=max(mx,query(L,R,mid+1,r,rt<<1|1));
    return mx;
}
bool check(LL x){
    int sz=pa.size();
    build(0,sz,1);
    update(0,sz,1,lower_bound(pa.begin(),pa.end(),0)-pa.begin(),0);
    LL sum=0;
    for(int i=1;i<=n;i++){
        sum=sum+(LL)a[i];
        int l=lower_bound(pa.begin(),pa.end(),sum-x)-pa.begin();
        int ge=query(l,sz,0,sz,1);
        if(ge==-1)continue;
        int z=lower_bound(pa.begin(),pa.end(),sum)-pa.begin();
        ge=max(ge+1,query(z,z,0,sz,1));
        update(0,sz,1,z,ge);
    }
    return query(0,sz,0,sz,1)>=k;
}
int main(){
    int T;cin>>T;
    while(T--){
        scanf("%d%d",&n,&k);
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        pa.clear(),pa.push_back(0);
        for(int i=1;i<=n;i++) pa.push_back(pa[i-1]+(LL)a[i]);
        sort(pa.begin(),pa.end());pa.erase(unique(pa.begin(),pa.end()),pa.end());
        LL l=-1e17,r=1e17,ans;
        while(l<=r){
            LL mid=(l+r)/2;
            if(check(mid)) ans=mid,r=mid-1;
            else l=mid+1;
        }
        cout<<ans<<'\n';
    }
    return 0;
}