HDU 6606. Distribution of books (DP+线段树)

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

题意：
n个数，可以选择前m(m自定)个数分成k块，问每块的数字和的最大值最小是多少。
n∈[1, 2e5]，k∈[1, n]

分析：
先求前缀和leftSum[]，然后对前缀和离散化；
对leftSum[]排序并去重得到pre[]；
设f[i]=前i个数最多能分成的块数；
二分答案，设答案为x；
则转移方程为f[i]=max(f[j])+1，leftSum[i]-leftSum[j]<=x,j<i
不能O(n^2)转移，想办法加速；
以离散化后的pre[]的rank为下标，f为权值建立线段树；
线段树需要实现单点修改、区间求最大值操作；
每次求f[i]时，令L=lower_bound(pre+1,pre+1+num,leftSum[i]-x)，其中num为pre[]离散化后数字个数，
然后只需求得线段树[L,num]区间内的最大值_max，将f[i]更新为=_max+1即可，此时别忘了用新得到的f[i]去update线段树。
代码中注意一些小细节要处理好。
总时间复杂度为O(n*(logn)^2).

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long llong;
const int tmax=2e5+5;
int n,k,num,_max[tmax*4];
llong a[tmax],pre[tmax];
map<llong,int> rrank;
int query(int o,int l,int r,int ql,int qr)
{
    if(ql>qr) return -tmax;
    if(ql<=l&&r<=qr) return _max[o];
    int mid=l+(r-l)/2,ans=-tmax;
    if(mid>=ql) ans=query(o*2,l,mid,ql,qr);
    if(mid+1<=qr) ans=max(ans,query(o*2+1,mid+1,r,ql,qr));
    return ans;
}
void update(int o,int l,int r,int pos,int val)
{
    int lc=o*2,rc=o*2+1,mid=l+(r-l)/2;
    if(l==r) _max[o]=val;
    else
    {
        if(pos<=mid) update(lc,l,mid,pos,val);
        else update(rc,mid+1,r,pos,val);
        _max[o]=max(_max[lc],_max[rc]);
    }
    return;
}
void build(int o,int l,int r)
{
    _max[o]=-tmax;
    if(l==r) return;
    int mid=l+(r-l)/2;
    build(o*2,l,mid);
    build(o*2+1,mid+1,r);
    return;
}
bool check(llong x)
{
    build(1,1,num);
    int f,tmp;
    for(int i=1;i<=n;i++)
    {
        int l=lower_bound(pre+1,pre+1+num,a[i]-x)-pre;
        int r=num;
        f=query(1,1,num,l,r)+1;
        if(a[i]-x<=0) f=max(f,1);
        if(f>=k) return true;
        if(f>0) update(1,1,num,rrank[a[i]],f);
    }
    return false;
}
void work()
{
    llong l=-200000000000001ll,r=200000000000001ll,mid;
    while(r-l>4)
    {
        mid=(l+r)>>1;
        bool ok=check(mid);
        if(ok) r=mid;
        else l=mid+1;
    }
    for(;l<=r;l++)
        if(check(l)) break;
    printf("%I64d\n",l);
    return;
}
int main()
{
    int kase;
    scanf("%d",&kase);
    while(kase--)
    {
        num=0;
        rrank.clear();
        scanf("%d%d",&n,&k);
        for(int i=1;i<=n;i++)
        {
            scanf("%I64d",&a[i]);
            pre[i]=pre[i-1]+a[i];
            a[i]=pre[i];
        }
        sort(pre+1,pre+1+n);
        num=unique(pre+1,pre+1+n)-pre-1;
        pre[num+1]=200000000000001ll;
        for(int i=1;i<=num;i++)
            rrank[pre[i]]=i;    //第i小
        work();
    }
    return 0;
}