Hdu 6856 Breaking Down News —— 权值线段树+DP

This way

题意：

现在有一个长度为n的数组，你要将其划分成一些连续的区间，每个区间的长度在l~r之间，每个区间的贡献是：
-1：这个区间的和小于0
0：这个区间的和等于0
1：这个区间的和大于0
问你最后总的贡献最大是多少

题解：

我想用几个set来维护…但是失败了，有点难写。确实用权值线段树的话会方便很多
dp[i]表示到了第i个位置的答案最大是多少
首先我们知道对于第i个位置，有最多三种转移的可能:dp[i]=dp[j]-1,dp[i]=dp[k],dp[i]=dp[l]+1
那么其实就对应三种情况：（设pre[i]表示1~i的a数组的和）
pre[j]>pre[i]，pre[k]=pre[i]，pre[l]<pre[i]
那么我们将所有前缀和离散化之后，以前缀和的值开权值线段树，然后三种情况查找转移一下即可

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+5,inf=1e8;
int mx[N*4];
void update(int l,int r,int root,int p,int v){
    if(l==r){
        mx[root]=v;
        return ;
    }
    int mid=l+r>>1;
    if(mid>=p)
        update(l,mid,root<<1,p,v);
    else
        update(mid+1,r,root<<1|1,p,v);
    mx[root]=max(mx[root<<1],mx[root<<1|1]);
}
int query(int l,int r,int root,int ql,int qr){
    if(l>=ql&&r<=qr)return mx[root];
    int mid=l+r>>1;
    int ans=-inf;
    if(mid>=ql)
        ans=query(l,mid,root<<1,ql,qr);
    if(mid<qr)
        ans=max(ans,query(mid+1,r,root<<1|1,ql,qr));
    return ans;
}
int pos[N],a[N],b[N],sum[N],fa[N],dp[N];
int finds(int x){return x==fa[x]?x:fa[x]=finds(fa[x]);}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--){
        int n,l,r;
        scanf("%d%d%d",&n,&l,&r);
        for(int i=1;i<=n+2;i++)fa[i]=i;
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]),a[i]+=a[i-1],b[i]=a[i];
        int all=n+1;
        b[all]=0;
        sort(b+1,b+1+all);
        for(int i=0;i<=n;i++){
            pos[i]=finds((int)(lower_bound(b+1,b+1+all,a[i])-b));
            fa[pos[i]]=finds(pos[i]+1);
        }
        for(int i=1;i<=all*4;i++)mx[i]=-inf;
        for(int i=1;i<=n;i++)dp[i]=-inf;
        for(int i=l;i<=n;i++){
            if(i-r>0)update(1,all,1,pos[i-r-1],-inf);
            update(1,all,1,pos[i-l],dp[i-l]);
            int pr=lower_bound(b+1,b+1+all,a[i])-b;
            if(pr>1)
                dp[i]=max(dp[i],query(1,all,1,1,pr-1)+1);
            int pl=upper_bound(b+1,b+1+all,a[i])-b;
            if(pl<=all)
                dp[i]=max(dp[i],query(1,all,1,pl,all)-1);
            dp[i]=max(dp[i],query(1,all,1,pr,pl-1));
        }
        printf("%d\n",dp[n]);
    }
    return 0;
}