Codeforces Global Round 10 H. ZS Shuffles Cards(期望线性性几何分布)

最新推荐文章于 2025-09-22 00:48:09 发布

原创最新推荐文章于 2025-09-22 00:48:09 发布 · 289 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#概率 #期望 #几何分布 #期望线性性

概率dp/期望/概率专栏收录该内容

29 篇文章

订阅专栏

探讨了在n+m张牌的牌堆游戏中，通过数学和编程方法计算游戏结束期望时间的问题。游戏规则包括随机抽取数字牌和王牌，以及王牌触发的牌堆重置条件。解析了期望时间的计算公式，提供了详细的解题思路和代码实现。

题目

牌堆上有n+m(n<=2e6,m<=2e6)张牌，n张是1到n的数字牌，m张是王牌，

初始时，n+m张牌被随机打乱放到牌堆上，主人公有一个初始为空的S集合，

每一秒，主人公都从牌堆顶抽一张牌，

（1）如果是数字牌，就会把这个数放到S集合中去，并把这张牌移除

（2）如果是王牌，

①如果S集合此时已经包含n个数，游戏结束

②否则，把所有牌都放回牌堆，注意S集合内已放入的数保持不变，

再把n+m张牌随机打乱形成新的牌堆，继续抽取，打乱不计时间

问最后游戏结束的期望时间

思路来源

codeforces comment

https://www.cnblogs.com/Hs-black/p/13518909.html

https://www.cnblogs.com/st1vdy/p/13518667.html#4662181

题解

首先，证一波独立，证明看不懂，直接抄过来

E(ans)=E(每一轮迭代的秒数)*E(迭代的轮数)，称每一次抽到王牌为一次迭代

根据期望线性性，

E(每一轮迭代的秒数)

=E(每一张数字牌在所有王牌的左边)+1

=E(1在所有王牌左边)+E(2在所有王牌左边)+...+王牌产生的1秒贡献

= $n*\frac{1}{m+1}+1$ ，1/(m+1)是只考虑这张数字牌和m张王牌

接下来求E(迭代的轮数)，记f(i)为还有i个数没搞到S里去时，到游戏结束时迭代轮数的期望

则对于f(1)来说，每次都是1/(m+1)的机会能搞进去，搞不进去就会一直重复，几何分布，期望为m+1

而对于f(i)来说，i/(m+i)概率搞到一张数字牌转移到f(i-1)，m/(m+i)搞到王牌浪费了一轮还需要f(i)轮，

$f(i)=\frac{i}{m+i}*f(i-1)+\frac{m}{m+i}*(f(i)+1)$ ，化简一下 $f(n)=f(1)+\sum_{i=2}^{n}\frac{m}{i}$

二者相乘即可，第一部分不用期望线性性的做法是O(n)的，

可参考https://www.cnblogs.com/st1vdy/p/13518667.html#4662181

代码

#include<bits/stdc++.h>
using namespace std;
const int mod=998244353;
int modpow(int x,int n,int mod){
    int res=1;
    for(;n;n>>=1,x=1ll*x*x%mod)
        if(n&1)res=1ll*res*x%mod;
    return res;
}
int inv(int x){
    return modpow(x,mod-2,mod);
}
int n,m,ans,sum;
int main(){
    scanf("%d%d",&n,&m);
    ans=(1ll*n*inv(m+1)+1)%mod;
    for(int i=1;i<=n;++i){
        sum=(sum+inv(i))%mod;
    }
    sum=(1ll*sum*m+1)%mod;
    printf("%d\n",1ll*ans*sum%mod);
    return 0;
}