Codeforces 1194 F Crossword Expert —— 概率dp+组合数

This way

题意：

给你n个问题解决所需要的时间，你会从1-n的去解决问题，每个问题有$\frac{1}{2}$的可能在$a_i$时间解决，剩下的可能在$a_i+1$的时间解决。为问你最终解决问题数量的期望。

题解：

做出有点水平的概率题目有点小激动。
因为它是依次过来解决这些问题的，那么做法应该是枚举每个位置，看看做到这里的概率是多少。假设到第i个位置为止，最小的解决时间是sum,最大解决时间是sum+i。那么当sum+i<=T的时候，能够解决这个问题的概率为1。那么如果sum>T的话，解决这个问题的概率是0，就可以结束了。
剩下的是中间这一段。首先假设res=T-sum，也就是能够容忍多少问题多出那个1，那么可以知道第i个位置能够过的概率是$C_i^0+C_i^1+...+C_i^{res}$，但是怎么维护这个东西呢，我在杨辉三角上找不到答案，所以应该是从另一个角度出发考虑，从dp的角度去看的话，设dp[i]表示到第i个位置成功的几率。那么假设上一个位置的res是pre，成功的概率是不是少了$C_{i-1}^{res+1}+...+C_{res}^{pre}$这么多，那么就减掉这些可能性。同时这个位置有两种可能，第一种可能的话，直接加上维护的这个值$\ast \frac{1}{2}$即可。如果是多1的话，那么再减掉$C_{i-1}^{res}\ast \frac{1}{2^{i-1}}$再乘上$\frac{1}{2}$即可。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=2e5+5;
const ll mod=1e9+7;
ll dp[N];
ll qpow(ll a,ll b){ll ans=1;for(;b;b>>=1,a=a*a%mod)if(b&1)ans=ans*a%mod;return ans;}
ll fac[N],inv[N];
ll c(int n,int m){
    if(n<m||m<0)return 0;
    if(n==m||m==0)return 1;
    return fac[n]*inv[m]%mod*inv[n-m]%mod;
}
int main()
{
    fac[0]=1;
    for(ll i=1;i<N;i++)
        fac[i]=fac[i-1]*i%mod,inv[i]=qpow(fac[i],mod-2);
    int n;
    ll T,x,sum=0;
    scanf("%d%lld",&n,&T);
    ll ans=0;
    int f=0;
    for(ll i=1;i<=n;i++){
        scanf("%lld",&x);
        sum+=x;
        if(sum+i<=T)
            dp[i]=1,ans=(ans+1)%mod;
        else if(sum>T)
            return 0*printf("%lld\n",ans);
        else{
            ll res=T-sum;
            if(!f){
                for(int j=0;j<=res;j++)
                    dp[i]=(dp[i]+c(i,j))%mod;
                dp[i]=dp[i]*qpow(qpow(2,i),mod-2)%mod;
                f=res;
            }
            else{
                dp[i]=dp[i-1];
                while(f>res)
                    dp[i]=(dp[i]-c(i-1,f--)*qpow(qpow(2,i-1),mod-2)%mod+mod)%mod;
                dp[i]=(dp[i]*qpow(2,mod-2)+(dp[i]-c(i-1,f)*qpow(qpow(2,i-1),mod-2)%mod+mod)*qpow(2,mod-2))%mod;
            }
            ans=(ans+dp[i])%mod;
        }
    }
    printf("%lld\n",ans);
    return 0;
}