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

Posted sdfzsyq

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了LUOGU P2344 奶牛抗议 (树状数组优化dp)相关的知识,希望对你有一定的参考价值。

 

 

 传送门

 

解题思路

树状数组优化dp,f[i]表示前i个奶牛的分组的个数,那么很容易得出$f[i]=sumlimits_{1leq jleq 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 = 100005;
const int mod = 1000000009;

inline int rd(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)) {f=ch==-?0:1;ch=getchar();}
    while(isdigit(ch))  {x=(x<<1)+(x<<3)+ch-0;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=0;
    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[0]=1;
    for(register int i=1;i<=n;i++) sum[i]=sum[i-1]+rd(),cpy[i]=sum[i];
    sort(cpy+1,cpy+1+n,cmp);int u=unique(cpy+1,cpy+1+n)-cpy-1;
    for(register int i=1;i<=n;i++) rk[i]=lower_bound(cpy+1,cpy+1+u,sum[i])-cpy;
    for(register int i=1;i<=n;i++){
        if(sum[i]>=0) f[i]=1;
        f[i]+=query(rk[i]);
        update(rk[i],f[i]);
    }
    cout<<f[n]<<endl;
    return 0;
}
View Code

 

以上是关于LUOGU P2344 奶牛抗议 (树状数组优化dp)的主要内容,如果未能解决你的问题,请参考以下文章

奶牛抗议 DP 树状数组

Luogu2345 奶牛集会(树状数组)

luogu P2345 奶牛集会 |排序+树状数组

奶牛抗议

luogu AT2300 Snuke Line |树状数组

树状数组1模板 Luogu 3374