Codeforces 1000D Yet Another Problem On a Subsequence

题意：

定义一个数列是“好的”：第一个数字a[0]为数列长度+1。

定义一个数列的子序列是“好的”：这个子序列能分割成几个“好的”数列。子序列中的每个元素都必须属于其中一个“好的”数列。

给出一个数列，求“好的”子序列的数目。

注意：是分割成 eg。1 1 2 2 就不能分成1 2, 1 2这样的两个好的数列，因为不是分割形成的

思路：

dp[i]是以i开头的“好的”子序列数目

①dp[i]=C( n-i,a[i])  ：从i后面任意挑a[i]个数字来构成“好的”子序列，“好的”子序列的长度为a[i]+1

②dp[i]+=C(j-i-1,a[i])*dp[j] ：从[i+1,j-i-1]中挑a[i]个 再从[j+1,n]中挑以j开头的“好的”子序列

 

#include <cstdio>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <set>
#include <iostream>
#define ll long long
#define INF 0x3f3f3f
using namespace std;
const int N=1000+100;
const int mod=998244353;
int F[N], Finv[N], inv[N]; //F是阶乘，Finv是阶乘的逆元
int unit;
ll res[N];
void init()
{
    F[0] = Finv[0] = inv[1] = 1;
    for (register int i = 2; i < N; i++)
        inv[i] = (mod - mod / i) * 1ll * inv[mod % i] % mod;
    for (register int i = 1; i < N; i++){
        F[i] = F[i - 1] * 1ll * i % mod;
        Finv[i] = Finv[i - 1] * 1ll * inv[i] % mod;
    }
}
int comb(int n, int m){///comb(n, m)?íê?C(n, m)
    if (m < 0 || m > n) return 0;
    return F[n] * 1ll * Finv[n - m] % mod * Finv[m] % mod;
}
ll a[N],dp[N],ans;
int main()
{
    int n,i,j;
    init();
    scanf("%d",&n);
    for(i=1;i<=n;i++) scanf("%lld",&a[i]);
    for(i=n;i>=1;i--)
    {
        if(i+a[i]>n || a[i]<=0) continue;
        dp[i]=comb(n-i,a[i]);
        for(j=a[i]+i+1;j<=n;j++)
        {
            dp[i]=(dp[i]+(comb(j-i-1,a[i])*dp[j])%mod)%mod;
        }
    }
    for(i=1;i<=n;i++)
        ans=(ans+dp[i])%mod;
    printf("%lld\n",ans);
}