CF1194F Crossword Expert（数论，组合数学）

不难的一题。不知道为什么能 $2500$……

不过场上推错了一直不会优化……

首先考虑 $f_i$ 表示恰好做完前 $i$ 道题的概率。

这样很难算。修改一下，$f_i$ 表示做完至少 $i$ 道题的概率。

答案就是 $\sum\limits_{i=0}^ni(f_i-f_{i+1})=\sum\limits_{i=1}^nf_i$。

由于每道题只可能多用至多一秒，考虑 $dp[i][j]$ 为前 $i$ 道题恰好SB $j$ 次的概率。

初始状态是 $dp[0][0]=1$。转移是 $dp[i][j]=\dfrac{1}{2}(f[i-1][j]+f[i-1][j-1])$。

盯着式子看不难看出 $dp[i][j]=(\dfrac{1}{2})^i\dbinom{i}{j}$。用实际意义也不难理解。

（场上就是这里推成了 $(\dfrac{1}{2})^{i+j}\dbinom{i+j}{i}$ 就自闭了……）

那么有 $f_i=\sum\limits_{j=0}^{r_i}dp[i][j]=(\dfrac{1}{2})^i\sum\limits_{j=0}^{r_i}\dbinom{i}{j}$。其中 $r_i=T-\sum\limits_{j=1}^it_j$，表示最多允许SB几次。（其实要和 $i$ 取个 $\min$，但是不影响，可以想一想为什么）

问题就是求 $\sum\limits_{j=0}^{r_i}\dbinom{i}{j}$ 了。接下来是一个很妙的做法。

首先 $i=1$ 时直接暴力。

然后 $\sum\limits_{j=0}^{r_i}\dbinom{i+1}{j}=\sum\limits_{j=0}^{r_i}(\dbinom{i}{j}+\dbinom{i}{j-1})=2\sum\limits_{j=0}^{r_i}\dbinom{i}{j}-\dbinom{i}{r_i}$。可以直接递推。

由于 $r_i$ 单调递减，递推完之后把 $r_{i+1}+1$ 到 $r_i$ 的组合数都删掉就行了。

实现优秀一点可以做到 $O(n)$。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=,mod=,inv2=;
#define FOR(i,a,b) for(int i=(a);i<=(b);i++)
#define ROF(i,a,b) for(int i=(a);i>=(b);i--)
#define MEM(x,v) memset(x,v,sizeof(x))
inline ll read(){
    ll x=,f=;char ch=getchar();
    while(ch<'' || ch>'') f|=ch=='-',ch=getchar();
    while(ch>='' && ch<='') x=x*+ch-'',ch=getchar();
    return f?-x:x;
}
int n,t[maxn],fac[maxn],inv[maxn],invfac[maxn],f[maxn],ans,pro=;
ll r[maxn];
int C(int n,ll m){
    if(n<m) return ;
    return 1ll*fac[n]*invfac[m]%mod*invfac[n-m]%mod;
}
int main(){
    n=read();r[]=read();
    FOR(i,,n) t[i]=read();
    FOR(i,,n) r[i]=r[i-]-t[i];
    fac[]=fac[]=inv[]=invfac[]=invfac[]=;
    FOR(i,,n){
        fac[i]=1ll*fac[i-]*i%mod;
        inv[i]=mod-1ll*(mod/i)*inv[mod%i]%mod;
        invfac[i]=1ll*invfac[i-]*inv[i]%mod;
    }
    FOR(i,,min(1ll,r[])) f[]=(f[]+C(,i))%mod;
    FOR(i,,n){
        if(r[i]<) break;
        f[i]=(2ll*f[i-]-C(i-,r[i-])+mod)%mod;
        ROF(j,min<ll>(i,r[i-]),r[i]+) f[i]=(f[i]-C(i,j)+mod)%mod;
    }
    FOR(i,,n){
        if(r[i]<) break;
        pro=1ll*pro*inv2%mod;
        ans=(ans+1ll*pro*f[i])%mod;
    }
    printf("%d\n",ans);
}