【离散+线段树】HDU - 6606 - Distribution of books

题目链接http://acm.hdu.edu.cn/showproblem.php?pid=6606

---

题意

有n个数字，取一个前缀，分成k块。要求每块的最大值最小，问最小是多少。

---

题解

官方题解：
题目要使得最大值最小？考虑二分答案
对于每次二分答案（假设为x），如何判定x能否满足分为k份的要求呢？考虑动态规划
令dp[i]表示前i个数最多能分成几段, 则：$dp[i]=max(dp[j])+1,(sum[i]-sum[j]<=x)$
如果直接dp，时间复杂度为n^2，显然会TLE!
考虑用平衡树维护或者离散化后权值线段树维护，总体复杂度n*log(n) n*log(n)*log(n)

瞄一眼题解，发现巨水系列。
我写了个离散线段树，维护区间的时候不需要二分，就把值当成参数传进去比较即可，可以少一个log

---

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+7;
const ll inf=3e14;
int T,n,k,tot;
int a[N];
ll sum[N],b[N];
int t[N<<2];

void bd(int rt,int l,int r){
    t[rt]=0;
    if(l==r) return;
    int m=l+r>>1;
    bd(rt<<1,l,m);
    bd(rt<<1|1,m+1,r);
}
void upd(int rt,int l,int r,ll x,int v){
    if(l==r){t[rt]=max(t[rt],v);return;}
    int m=l+r>>1;
    if(x<=b[m]) upd(rt<<1,l,m,x,v);
    if(b[m]<x) upd(rt<<1|1,m+1,r,x,v);
    t[rt]=max(t[rt<<1],t[rt<<1|1]);
}
int que(int rt,int l,int r,ll x){
    if(x<=b[l]) return t[rt];
    if(l==r) return 0;
    int m=l+r>>1,ret=0;
    if(x<=b[m]) return max(que(rt<<1,l,m,x),t[rt<<1|1]);
    return que(rt<<1|1,m+1,r,x);
}
bool ck(ll x){
    bd(1,1,tot);
    for(int i=1;i<=n;i++){
        int tmp=0;
        if(sum[i]<=x) tmp=1;
        int mx=que(1,1,tot,sum[i]-x);
        if(mx) tmp=mx+1;
        if(tmp>=k) return true;
        if(tmp) upd(1,1,tot,sum[i],tmp);
    }
    return false;
}

int main(){
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&k);
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            sum[i]=sum[i-1]+a[i];
            b[i]=sum[i];
        }
        sort(b+1,b+1+n);
        tot=unique(b+1,b+1+n)-b-1;
        ll lo=-inf,hi=inf,ans=-1;
        while(lo<=hi){
            ll mid=(lo+hi)/2;
            if(ck(mid)) hi=mid-1,ans=mid;
            else lo=mid+1;
        }
        printf("%lld\n",ans);
    }
}