CodeForces 1194F Crossword Expert （期望+组合数学）

题目链接：点击这里

题目大意：
有 $n$ 个任务，每个任务需要 $t_i$ 或 $t_i+1$ 的时间完成，要按给出的顺序解决，求在 $T$ 时间内，完成任务数的期望

题目分析：
设解决第 $i$ 个任务的概率为 $P_i$ ，完成任务数的数期望为 $E$ ，由题意可得：
$$E=\sum _{i=1}^{n}i{P}_i$$
$$=P_1+P_2+P_2+P_3+P_3+P_3+...$$
设 $f_i$ 为最终完成任务数大于等于 $i$ 的概率，有：
$$f_i=\sum _{k=i}^n{P}_k$$
则此时有：
$$E=\sum _{i=1}^n{f}_i$$
接下来我们分类讨论一些情况：
设$su{m}_i=\sum _{k=1}^i{t}_i$ ，则前 $i$ 个任务花费的最少时间为 $su{m}_i$ ，最长时间为 $su{m}_i+i$

1. $su{m}_i>T$ 时，说明不可能完成，应在此时 $break$
2. $su{m}_i+i\le T$ ，说明此时一定能完成，此时有 $f_i=1$
3. $su{m}_i\le T<su{m}_i+i$ ，此时至多有 $T-su{m}_i$ 个任务多花费了 $1$ 分钟，因为我们枚举 $1$ 到 $T-su{m}_i$ 用组合数就能统计出所有可能的方案数，而总方案数为 $2^i$ ，所以有：
   $$f_i=\frac{{\sum }_{k=0}^{T-su{m}_i}{C}_i^k}{2^i}$$

所以有：
$$E=\sum _{i=1}^n\frac{{\sum }_{k=0}^{T-su{m}_i}{C}_i^k}{2^i}$$
每次计算 $f_i$ 用循环是会超时的，我们考虑优化组合数部分：
我们知道：
$$C_n^m+c_n^{m+1}=C_{n+1}^{m+1}$$
于是可以写这样一组方程：
$$\{\begin{array}{c}{C}_{n+1}^0=C_n^0\\ C_{n+1}^1=C_n^0+C_n^1\\ C_{n+1}^2=C_n^1+C_n^2\\ ...\\ C_{n+1}^{m-1}=C_n^{m-2}+C_n^{m-1}\\ C_{n+1}^m=C_n^{m-1}+C_n^m\end{array}$$
两边累加可以得到：
$$\sum _{k=0}^m{C}_{n+1}^k=2\sum _{k=0}^m{C}_n^k-C_n^m$$
因为 $T-su{m}_i$ 是单调递减的，我们可以直接用上述公式又上一层转移到下一层

具体细节见代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<set>
#include<map>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
ll read()
{
	ll res = 0,flag = 1;
	char ch = getchar();
	while(ch<'0' || ch>'9')
	{
		if(ch == '-') flag = -1;
		ch = getchar();
	}
	while(ch>='0' && ch<='9')
	{
		res = (res<<3)+(res<<1)+(ch^48);//res*10+ch-'0';
		ch = getchar();
	}
	return res*flag;
}
const int maxn = 2e5+5;
const int mod = 1e9+7;
const double pi = acos(-1);
const double eps = 1e-8;
ll n,T,t[maxn],fac[maxn],inv[maxn];
ll qpow(ll a,ll b)
{
	ll res = 1;
	while(b)
	{
		if(b&1) res = res*a%mod;
		a = a*a%mod;
		b >>= 1;
	}
	return res;
}
void init()
{
	fac[0] = 1;
	for(int i = 1;i < maxn;i++)
		fac[i] = fac[i-1]*i%mod;
	inv[maxn-1] = qpow(fac[maxn-1],mod-2);
	for(int i = maxn-1;i;i--)
		inv[i-1] = inv[i]*i%mod;
} 
ll C(int n,int m)
{
	if(m==0 || n==m) return 1;
	return fac[n]*inv[m]%mod*inv[n-m]%mod;
}
ll sigma,last;
ll get_res(int pos,ll rem)
{
	if(sigma == 0)
	{
		sigma = 1;
		for(int i = 1;i <= rem;i++)
			sigma = (sigma+C(pos,i))%mod;
		last = rem;
		return sigma;
	}
	sigma = (sigma*2-C(pos-1,last)+mod)%mod; //先转移
	for(int i = last;i >= rem+1;i--) //再把多余的减去
		sigma = (sigma-C(pos,i)+mod)%mod;
	last = rem;
	return sigma;
}
int main()
{
	init();
	n = read(),T = read();
	for(int i = 1;i <= n;i++)
		t[i] = read();
	ll sum = 0,base = 1,ans = 0;
	for(int i = 1;i <= n;i++)
	{
		sum += t[i];
		base = base*2%mod;
		if(T-sum >= i)
		{
			ans = (ans+1)%mod;
			continue;
		}
		if(T-sum < 0) break;
		ans = (ans+get_res(i,T-sum)*qpow(base,mod-2)%mod)%mod;
	}
	printf("%lld\n",ans);
	return 0;
}