题意:
给你n个问题解决所需要的时间,你会从1-n的去解决问题,每个问题有 1 2 \frac{1}{2} 21的可能在 a i a_i ai时间解决,剩下的可能在 a i + 1 a_i+1 ai+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
r
e
s
C_{i}^{0}+C_{i}^{1}+...+C_{i}^{res}
Ci0+Ci1+...+Cires,但是怎么维护这个东西呢,我在杨辉三角上找不到答案,所以应该是从另一个角度出发考虑,从dp的角度去看的话,设dp[i]表示到第i个位置成功的几率。那么假设上一个位置的res是pre,成功的概率是不是少了
C
i
−
1
r
e
s
+
1
+
.
.
.
+
C
r
e
s
p
r
e
C_{i-1}^{res+1}+...+C_{res}^{pre}
Ci−1res+1+...+Crespre这么多,那么就减掉这些可能性。同时这个位置有两种可能,第一种可能的话,直接加上维护的这个值
∗
1
2
*\frac{1}{2}
∗21即可。如果是多1的话,那么再减掉
C
i
−
1
r
e
s
∗
1
2
i
−
1
C_{i-1}^{res}*\frac{1}{2^{i-1}}
Ci−1res∗2i−11再乘上
1
2
\frac{1}{2}
21即可。
#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;
}