奶牛抗议

题目背景

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

说明

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

#include<bits/stdc++.h>
#define REP(i, a, b) for(int i = (a); i <= (b); ++ i)
#define REP(j, a, b) for(int j = (a); j <= (b); ++ j)
#define PER(i, a, b) for(int i = (a); i >= (b); -- i)
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
const ll mod=1000000009;
template <class T>
inline void rd(T &ret){
    char c;
    ret = 0;
    while ((c = getchar()) < '0' || c > '9');
    while (c >= '0' && c <= '9'){
        ret = ret * 10 + (c - '0'), c = getchar();
    }
}
queue<int>q;
int vis[maxn],p[maxn],n,dp[maxn];
int main()
{
    q.push(0);
    dp[0]=1;
    rd(n);
    REP(i,1,n)scanf("%d",&p[i]);
    while(!q.empty()){
        int cur=q.front();
        q.pop();
        ll tmp=0;
        REP(i,cur+1,n){
            tmp+=p[i];
            if(tmp>=0){
                dp[i]=(dp[i]+dp[cur])%mod;
                if(!vis[i]){
                    q.push(i);
                    vis[i]=1;
                }
            }
        }
    }
    cout<<dp[n]<<endl;
    return 0;
}
原文地址:https://www.cnblogs.com/czy-power/p/10455792.html