P2344 奶牛抗议

题目背景

Generic Cow Protests, 2011 Feb
题目描述

约翰家的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

说明

解释:如果分两组，可以把前三头分在一组，或把后三头分在一组；如果分三组，可以把中间两头分在一组，第一和最后一头奶牛自成一组；最后一种分法是把四头奶牛分在同一组里。

Code:
#include
#include
#include
#include
#include
#include
#include
using namespace std;
const int N=100005,MOD=1000000009;
int n,sum[N],c[N],a[N],ans;
int lowbit(int i){
    return i&(-i);
}
void update(int i,int val){
    while(i<=n+1){
        c[i]=(c[i]+val)%MOD;
        i+=lowbit(i);
    }
}
int get_sum(int i){
    int res=0;
    while(i>0){
        res=(res+c[i])%MOD;
        i-=lowbit(i);
    }
    return res;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++) {
        scanf("%d",&sum[i]);
        sum[i]+=sum[i-1];
        a[i]=sum[i];
    }
    sort(a,a+1+n);
    for(int i=0;i<=n;i++){
        sum[i]=lower_bound(a,a+1+n,sum[i])-a+1;
    }
    update(sum[0],1);
    for(int i=1;i<=n;i++){
        ans=get_sum(sum[i]);
        update(sum[i],ans);
    }
    printf("%d\n",ans);
    return 0;
}

转载于:https://www.cnblogs.com/ukcxrtjr/p/11206778.html