LUOGU P2344 奶牛抗议 (树状数组优化dp)

传送门

解题思路

树状数组优化dp，f[i]表示前i个奶牛的分组的个数，那么很容易得出$f[i]=\sum\limits_{1\leq j\leq i}f[j-1]*(sum[i]\ge sum[j-1])$，但是这样的时间复杂度是$O(n^2)​$，所以考虑优化，发现必须满足$sum[i]\ge sum[j-1]​$才能进行转移，那么直接离散化后用树状数组维护一个前缀和即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>

using namespace std;
const int MAXN = ;
const int mod = ;

inline int rd(){
    int x=,f=;char ch=getchar();
    while(!isdigit(ch)) {f=ch=='-'?:;ch=getchar();}
    while(isdigit(ch))  {x=(x<<)+(x<<)+ch-'';ch=getchar();}
    return f?x:-x;
}

//f[i] 到了第i个人
//f[i]=f[i-k]+1 

int f[MAXN],t[MAXN],rk[MAXN];
int n,sum[MAXN],cpy[MAXN];

inline bool cmp(int x,int y){
    return x<y;
}

int query(int x){
    int ret=;
    for(;x;x-=x&-x) ret+=t[x],ret%=mod;
    return ret;
}

void update(int x,int k){
    for(;x<=n;x+=x&-x) t[x]+=k,t[x]%=mod;
}

int main(){
    n=rd();f[]=;
    for(register int i=;i<=n;i++) sum[i]=sum[i-]+rd(),cpy[i]=sum[i];
    sort(cpy+,cpy++n,cmp);int u=unique(cpy+,cpy++n)-cpy-;
    for(register int i=;i<=n;i++) rk[i]=lower_bound(cpy+,cpy++u,sum[i])-cpy;
    for(register int i=;i<=n;i++){
        if(sum[i]>=) f[i]=;
        f[i]+=query(rk[i]);
        update(rk[i],f[i]);
    }
    cout<<f[n]<<endl;
    return ;
}