2019牛客暑期多校训练营（第八场）J.Just Jump(dp前缀和优化+组合数容斥)

题意

有一条长为$L$的河，你在位置$0$你要通过这条长为$L$的河到达$L$，河中从$1,2,3,\cdots ,L-1$有石子可以踩上去通过，你每一次所走的距离必须要大于等于$d$，并且还存在$m$次攻击，每一次攻击由二元组$(t_i,p_i)$组成，表示，在$t_i$次跳跃时落在$p_i$时你就会被攻击，若被攻击你就会失败，求从$0$出发到达$L$的方法数

思路

先考虑没有攻击时总的方案数，有
$$f[i]=f[1]+f[2]+f[3]+\cdots +f[i-d]$$
$$f[i+1]=f[1]+f[2]+f[3]+\cdots +f[i-d]+f[i+1-d]$$
可以发现我们要是每次维护一个前缀和就可以$O(1)$求得的$f[i]$
$$f[i]=sum[i-1-d]+f[i-d]$$
所以可以在$O(L)$的时间求的总的方法数$f[L]$
那么最终的答案我们可以考虑用$f[L]$去减去被攻击时的方案数，就是最终的答案。
我们考虑一下每一个$(t,p)$的贡献，可以分成两部分考虑

1. 从$0$跳$t$次到达$p$，那么我们先考虑没有每次至少跳$d$步的限制，那么就是从$0$到$p$，跳$t$次，每次可以跳任意步数。转换一下可以变为有$p$个球，放到$t$个箱子里，箱子可以为空的方案数，那就可以套用经典的隔板法那么答案就是$C_{p+t-1}^{t-1}$。现在有了每次至少跳$d$的限制那么我们可以在一开始$p-dt$，那么剩下的这段距离就是原来的无限制的方案数了，那么方案数就是$$C_{p-dt+t-1}^{t-1}$$
2. 从$p$跳到$L$的方案数，这段的方案数其实和从$0$跳到$L-p$是等价，那么方案数就是$f[L-p]$

那么每一个攻击的贡献就是这两部分的方案数之积了，解决了某一个攻击的贡献，我们还要考虑一下攻击和攻击之间的关系。我们可以先把攻击按照跳的次数排个序，由于题目中跳跃的次数是没有重复的，我们通过排序后，令$dp[i]$为跳$t_i$次第一次被攻击时的方案数，那么有转移方程
$$dp[i]=C_{p_i-d{t}_i+t_i-1}^{t_i-1}-\sum _{j=0}^{i-1}dp[j]\ast C_{p_i-pj-d(t_i-t_j)+t_i-t_j-1}^{t_i-t_j-1}$$
$dp[i]$的方案数是从$0$到$p_i$跳$t_i$次的方案数减去，第$j$($0\le j<i$)次攻击作为第一次被攻击时的方案数乘上从第$j$次攻击跳到第$i$次攻击的方案数，就是第$i$次攻击作为第一次被攻击时的方案数了。
那么最终的答案就是
$$ans=f[L]-\sum _{i=0}^{m-1}dp[i]\ast f[L-p_i]$$
组合数预处理处理到1e7就好了但是，可能$d\ast t_i$可能会爆int，都改成long long 保险一点

#include<bits/stdc++.h>
using namespace std;
const int N=1e7+1234;
const int mod=998244353;
struct node
{
    int t,p;
}a[3005];
long long f[N],sum[N];
long long fac[N],inv[N];
long long dp[N];
long long quickmod(long long a,long long b)
{
    long long ans=1;
    while(b)
    {
        if(b%2==1)
            ans=ans*a%mod;
        a=a*a%mod;
        b=b/2;
    }
    return ans;
}
long long C(long long n,long long m)
{
    if(n<0||m<0) return 0;
    if(n<m)
        return 0;
    return fac[n]*inv[m]%mod*inv[n-m]%mod;
}
bool cmp(node a,node b)
{
    return a.t<b.t;
}
int main()
{
    fac[0]=1;
    for(int i=1;i<N;i++)
        fac[i]=fac[i-1]*i%mod;
    inv[N-1]=quickmod(fac[N-1],mod-2);
    for(int i=N-2;i>=0;i--)
        inv[i]=inv[i+1]*(i+1)%mod;
    int L,d,m;
    scanf("%d%d%d",&L,&d,&m);
    f[0]=sum[0]=1;
    for(int i=1; i<=L; i++)
    {
        if(i<d)
            f[i]=0;
        else
        {
            f[i]=(sum[i-1-d]+f[i-d])%mod;
        }
            sum[i]=(sum[i-1]+f[i])%mod;
    }
    for(int i=0;i<m;i++)
        scanf("%d%d",&a[i].t,&a[i].p);
    sort(a,a+m,cmp);
    long long ans=f[L];
    for(int i=0;i<m;i++)
    {
        dp[i]=C(a[i].p-1ll*a[i].t*d+a[i].t-1,a[i].t-1);
        for(int j=0;j<i;j++)
        {
            dp[i]=(dp[i]-dp[j]*C(a[i].p-a[j].p-1ll*d*(a[i].t-a[j].t)+a[i].t-a[j].t-1,a[i].t-a[j].t-1)%mod)%mod;
        }
        ans=(ans-dp[i]*f[L-a[i].p]%mod+mod)%mod;
    }
    printf("%lld\n",ans);
    return 0;
}