省选专练USACO11FEBGOLD奶牛抗议

水题。

由于之前存在一个前缀和和大于某个值的条件

直接树状数组优化罢了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef int INT;
#define int long long
const int mod=1e9+9;
const int N=2e6;
int maxn=2e5+100;
struct BIT{
    int T[N];
    inline int lowbit(int x){
        return x&(-x);
    }
    inline void update(int val,int x){
        while(x<=maxn){
            T[x]=(T[x]+val)%mod;
            x+=lowbit(x);
        }
    }
    inline int query(int x){
        int ret=0;
        while(x){
            ret+=T[x];
            ret%=mod;
            x-=lowbit(x);
        }
        return ret;
    }
}T;
int a[N];
int sum[N];
int p[N];
int f[N];
int n;
INT main(){
    scanf("%d",&n);
    sum[0]=0;
    for(int i=1;i<=n;i++){
        scanf("%lld",&a[i]);
        sum[i]=sum[i-1]+a[i];
        p[i]=sum[i];
    }
    p[0]=0;
//	if(sum[n]<0){
//		cout<<"Impossible";
//		return 0;
//	}
//	cout<<"working"<<endl;
//	for(int i=0;i<=n;i++){
//		cout<<a[i]<<" ";
//	}
    sort(p,p+1+n);
    int len=unique(p,p+1+n)-p-1;
    for(int i=0;i<=n;i++)
        sum[i]=lower_bound(p,p+1+len,sum[i])-p+1;
    memset(f,0,sizeof(f));
//	cout<<sum[0]<<endl;
    T.update(1,sum[0]);
//	cout<<"working"<<endl;
//	for(int i=0;i<=n;i++){
//		cout<<sum[i]<<" ";
//	}
    for(int i=1;i<=n;i++){
        f[i]=T.query(sum[i])%mod;
        T.update(f[i],sum[i]);
    }
    cout<<f[n]%mod;
}