2274: [Usaco2011 Feb]Generic Cow Protests

题目链接

题目大意：给出n个数，问有几种划分方案（同一个组内的数连续），使得每组数的和大于等于0

$题解：f[i]表示前i个数的分组方案数，根据经典转移方法，可得f[i]=\sum f[j]~|~sum(i)-sum(j)\ge 0$

$移项后就是sum(i)\ge sum(j)，一维偏序，这个明显不能排序，所以上树状数组$

$因为sum(i)可能很大，离散化一下$

注意树状数组下标不能为0，所以离散化后编号统一+1
j是从0开始枚举的，所以加上a[0]元素……

我的收获：naiev

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;

const int M=1000000;
#define P 1000000009

int n;
long long a[M+5],z[M+5];

struct array_tree{
    long long c[M+5];
    void updata(int x,long long v){
        for(int i=x;i<=n;i+=i&(-i))
            c[i]+=v,c[i]%=P;
    }
    long long query(int x){
        long long ret=0;
        for(int i=x;i>0;i-=i&(-i))
            ret+=c[i],ret%=P;
        return ret;
    }       
}T;

void work()
{
    T.updata(a[0],1);
    long long ans;
    for(int i=1;i<=n;i++){
        ans=T.query(a[i]);
        T.updata(a[i],ans);
    }
    cout<<ans<<endl;
}

void lishan()
{
    for(int i=0;i<=n;i++) z[i]=a[i];
    sort(z,z+1+n);
    int cnt=unique(z,z+1+n)-z-1;
    for(int i=0;i<=n;i++) a[i]=lower_bound(z,z+1+cnt,a[i])-z+1;//后移1位处理0 
}

void init()
{
    cin>>n;
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]),a[i]+=a[i-1];
    lishan();
}

int main()
{
    init();
    work();
    return 0;
}