2021牛客多校#9 I-Incentive Model（数学）

题目链接

传送门

题目大意（英语阅读理解）

有 $A、B$ 两人一起逐个挖 $n(1\le n<998244353)$块矿，对于同一块矿，有 $a$ 的概率属于 $A$ ，有 $b$ 的概率属于 $B$ ，其中 $a,b$为两方当时的股权占比， $a+b=1$。某方拥有一块矿后，其股权增加 $w$。初始 $A$ 拥有 $\frac{x}{y}$ 的股权，$B$ 拥有 $1-\frac{x}{y}$ 的股权。求 $A$ 能获得矿的期望数。

题解

设$S_i$表示第$i$块矿挖出后的股权。
由题意得，每获得一块后增加的期望为$\frac{w\ast S_{i-1}}{1+w(i-1)}$
可得递推式$$S_i=S_{i-1}+\frac{w\ast S_{i-1}}{1+w(i-1)}$$
但由于 $n$ 的值很大，故无法递推计算，我们考虑一下它的通项式。
可以转化为乘法，即$$\begin{array}{r}{S}_i=S_{i-1}\ast \frac{1+w\ast i}{1+w(i-1)}\end{array}$$
将其展开为$$S_i=\frac{1+w\ast i}{1+w(i-1)}\ast \frac{1+w\ast (i-1)}{1+w(i-2)}\ast ...\ast \frac{1+w\ast 1}{1+w\ast 0}$$
将其化简为$a(1+w\ast i)$
而$\frac{S_i-a}{w}=a\ast i$
所以答案为$n\ast a=n\frac{x}{y}$
用逆元计算即可。

参考代码

#include<bits/stdc++.h>
using namespace std;
const int mod=998244353;
int pow(int a,int b)
{
    int ret=1;
    while(b)
    {
        if(b&1)
            ret=1ll*ret*a%mod;
        a=1ll*a*a%mod;
        b>>=1;
    }
    return ret;
}
int main()
{
    int n,w,x,y;
    scanf("%d %d %d %d",&n,&w,&x,&y);
    printf("%d\n",1ll*n*x%mod*pow(y,mod-2)%mod);
}