题目
给定n(1<=n<=1e3)个数,第i个数是ai,一个好的序列定义为序列长度为a1+1,开头的正数是a1
求由多少个子序列是好的序列
思路来源
https://codeforces.com/blog/entry/60288
题解
子序列个数,就是任取字母的方案数即组合数啊
倒序考虑i,a[i]<=0跳过,dp[i]记录开头是i的合法方案数
后接另一个序列的时候,枚举后面接的好序列的起始位置是j
在[i+1,j-1]里取ai个,加上C(num,ai)*dp[j]即可
注意后面不接序列的情形,为统一,dp[n+1]=1
代码
#include<bits/stdc++.h>
using namespace std;
const int N=1e3+10;
const int mod=998244353;
//dp[i] 以i开头的方案数
int n,a[N],c[N][N],dp[N];
int sum;
int main()
{
scanf("%d",&n);
for(int i=0;i<=n;++i)
{
c[i][0]=c[i][i]=1;
for(int j=1;j<i;++j)
c[i][j]=(c[i-1][j]+c[i-1][j-1])%mod;
}
for(int i=1;i<=n;++i)
scanf("%d",&a[i]);
dp[n+1]=1;
for(int i=n;i>=1;--i)
{
if(a[i]<=0)continue;
for(int j=i+a[i]+1;j<=n+1;++j)//枚举下一段的位置
(dp[i]+=1ll*c[j-1-i][a[i]]*dp[j]%mod)%=mod;
}
for(int i=1;i<=n;++i)
(sum+=dp[i])%=mod;
printf("%d\n",sum);
return 0;
}