P2344 奶牛抗议

题目描述

约翰家的N 头奶牛正在排队游行抗议。一些奶牛情绪激动，约翰测算下来，排在第i 位的奶牛的理智度为Ai，数字可正可负。

约翰希望奶牛在抗议时保持理性，为此，他打算将这条队伍分割成几个小组，每个抗议小组的理智度之和必须大于或等于零。奶牛的队伍已经固定了前后顺序，所以不能交换它们的位置，所以分在一个小组里的奶牛必须是连续位置的。除此之外，分组多少组，每组分多少奶牛，都没有限制。

约翰想知道有多少种分组的方案，由于答案可能很大，只要输出答案除以1000000009 的余数即可。

输入输出格式

输入格式：

• 第一行：单个整数N，1 ≤ N ≤ 100000

• 第二行到第N + 1 行：第i + 1 行有一个整数Ai，−10^5 ≤ Ai ≤ 10^5

输出格式：

单个整数：表示分组方案数模1000000009 的余数

输入输出样例

输入样例#1：

4
2
3
-3
1

输出样例#1：

4

【解题思路】：

设d[i]表示前i头奶牛的分法，则有转移方程式如下：

 d[i]=sigma{ d[j]( j<i && S(j+1,i)>=0 ) }

【AC代码】：

#include<bits/stdc++.h>
#define M(a,b) memset(a,b,sizeof(a))
#define INF 0x3f3f3f3f
#define MOD 1000000009
using namespace std;
inline void read(int &x){
    char ch=getchar(),c=ch;
	x=0;
    while(ch<'0' || ch>'9'){
    	 c=ch;
		 ch=getchar();
	}
    while(ch>='0' && ch<='9'){
    	x=(x<<1)+(x<<3)+ch-'0';
		ch=getchar();
	}
    if(c=='-')x=-x;
}

int sum[100005],a[100005];
int hash[100005],cnt;
int n,i;

int C[100005],Max;
int lowbit(int x) { return x&(-x); }
int Sum(int x) {
    x++;
    int res=0;
    while(x>0) {
        res = (res+C[x])%MOD;
        x-=lowbit(x);
    }
    return res;
}
void Add(int x,int v) {
    x++;
    while(x<=Max+1) {
        C[x] = (C[x]+v)%MOD;
        x+=lowbit(x);
    }
}

int find(int x) {
    return lower_bound(hash,hash+cnt+1,x)-hash;
}

int main() {
    read(n);
    for(i=1;i<=n;i++){
        read(a[i]);
        sum[i]=sum[i-1]+a[i];
        Max=max(Max,sum[i]);
    }
    sort(sum,sum+n+1);
    for(i=1;i<=n;i++)
		if(sum[i]!=sum[i-1])
        	hash[++cnt]=sum[i];
    Add(find(0),1);
    int tot=0,ans=0;
    for(i=1;i<=n;i++){
        tot += a[i];
        ans = Sum(find(tot))%MOD;
        Add(find(tot),ans);
    }
    printf("%d\n",ans);
    return 0;
}