题解:
考试时想成了二维DP,没打出来,于是打了DFS。
首先考虑朴素DP,令F[i]表示以i结尾的分组方案,转移方程为 ,当且仅当 S[i] - S[j-1] >= 0 即S[i] >= S[j-1],S为前缀和。
对于一个i,我们要求出之前S[j-1]满足小于等于S[i]的F[j]的和,可以用树状数组维护。先把前缀和离散化,然后1到n转移,每次询问在i之前的满足条件的F[j]的和,并更新树状数组。
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<vector>
#include<cctype>
#include<climits>
#include<iomanip>
#define MAXA 100500
#define MOD 1000000009
using namespace std;
int T[MAXA],n;
void Add(int x,int y) {
for(;x<=n+7;x+=x & -x)
T[x] = (T[x] + y) % MOD;
}
int Ask(int x) {
int Ret = 0;
for(;x;x-=x & -x)
Ret = (Ret + T[x]) % MOD;
return Ret;
}
int a[MAXA],Pre[MAXA],Ans;
int main() {
scanf("%d",&n);
for(int i=1;i<=n;i++) {
scanf("%d",&a[i]);
a[i] += a[i-1];
Pre[i] = a[i];
}
sort(a,a + n + 1);
for(int i=0;i<=n;i++)
Pre[i] = lower_bound(a,a + n + 1,Pre[i]) - a + 2;
Add(Pre[0],1);
for(int i=1;i<=n;i++) {
Ans = Ask(Pre[i]);
Add(Pre[i],Ans);
}
printf("%d",Ans);
}