bzoj2274[Usaco2011 Feb]Generic Cow Protests

题意：给出n个数，问有几种划分方案（不能改变数的位置），使得每组中数的和大于等于0。输出方案数除以 1000000009的余数。

n^2dp很好想就不说了。
f[x]=f[x]+f[y] (sum[y]-sum[x-1]>=0)
把条件移项sum[y]>=sum[x-1]
那么我们可以把前缀和离散以后用树状数组求和。
树状数组优化dp。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=5e5+5;
typedef long long ll;
const int mo=1e9+9;
int sum[N],n;
struct node
{
    int s,id;
}a[N];
int t[N],b[N],f[N];
bool cmp(node a,node b)
{
    return (a.s<b.s)||(a.s==b.s&&a.id<b.id);
}
inline void add(int x,int y)
{
    while (x<=n)
    {
        t[x]=(t[x]+y)%mo;
        x+=x&-x;
    }
}
inline int get(int x)
{
    int ans=0;
    while (x>0)
    {
        ans=(ans+t[x])%mo;
        x-=x&-x; 
    }
    return ans;
}
int main()
{
    scanf("%d",&n);
    fo(i,1,n)
    {
        int x;
        scanf("%d",&x);
        a[i].id=i;
        a[i].s=a[i-1].s+x;
        f[i]=(a[i].s>=0);
    }
    sort(a+1,a+1+n,cmp);
    fo(i,1,n)b[a[i].id]=i;
    fo(i,1,n)add(b[i],f[i]=(f[i]+get(b[i]))%mo);
    printf("%d\n",f[n]);
    return 0;
}