CF1523E (概率+组合数学)

题目大意：

$n$盏灯，每次随机从熄灭的灯中选择一盏变亮，当连续有$k$个灯大于一盏亮着，则结束，求最终期望开灯次数

解题思路：

• 设$p_i$为第$i$次就结束的期望，$ans={\sum }_{i=1}^n{p}_i\cdot i$

• $ans={\sum }_{i=1}^n{p}_i\cdot i={\sum }_{i=1}^n{\sum }_{j=i}^n{p}_j$

• 对于${\sum }_{i=1}^n{p}_i\cdot i=p_1+2p_2+...+n{p}_n=(p_1+...+p_n)+(p_2+...+p_n)+...+(p_n)={\sum }_{i=1}^n{\sum }_{j=i}^n{p}_j$

• 令$g_i={\sum }_{j=i+1}^n{p}_j$

• $g_i$的含义就是，选择了$i$次还没有结束的期望

• 所以最后答案就是：${\sum }_{i=1}^{n-1}{g}_i$

• 那么$g_i$如何算呢，利用隔板法：
  $$\begin{gathered} 选择了i次还没结束等价n个空中选了i个空放球，这i个球相邻之间的距离至少为k-1 \\ 因为有i个空被选择还剩n-i个空未被选择，而i个球之间的空隙有i-1个，再加上两边的空袭有i+1个空隙 \\ 写成式子如下： \\ a+x_1+x_2+...+x_{i-1}+b=n-i \\ a\ge 0,b\ge 0,x_j\ge k-1(j\in [1,i-1]) \\ 所以就相当于先：n-i+2 \\ 然后：n-i+2-(i-1)\ast (k-2)从这里面选i个空,隔板一下就好 \\ 合法方案数为：C(n-(i-1)\ast (k-1),i) \\ {g}_i=\frac{C(n-(i-1)\ast (k-1),i)}{C(n,i)} \end{gathered}$$

AC代码：

#include <bits/stdc++.h>
#define ft first
#define sd second
#define pb push_back
#define IOS ios::sync_with_stdio(false), cin.tie(0), cout.tie(0) //不能跟puts混用
#define seteps(N) fixed << setprecision(N)
#define endl "\n"
const int maxn = 1e5 + 10;
using namespace std;
typedef long long ll;
typedef double db;
typedef pair<int, int> pii;
const ll mod = 1e9 + 7;
int t, n, k;
ll 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 - 2; i >= 0; i--) inv[i] = inv[i + 1] * (i + 1) % mod;
}
ll C(ll n, ll m) {
    if (m > n) return 0;
    return fac[n] * inv[m] % mod * inv[n - m] % mod;
}
int main() {
    init();
    cin >> t;
    while (t--) {
        ll ans = 1;
        cin >> n >> k;
        for (int i = 1; i <= n; i++) 
            ans = (ans + C(n - 1ll * (i - 1) * (k - 1), i) * qpow(C(n, i), mod - 2) % mod) % mod;
        cout << ans << endl;
    }
    return 0;
}