2021牛客暑期多校训练营#9:I-Incentive Model

2021牛客暑期多校训练营#9:I-Incentive Model

原题链接：https://ac.nowcoder.com/acm/contest/11260/I

题目大意

有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能获得矿的期望。

解题思路

一个接近$10^{10}$很明显不能用枚举法做，只能用通项式或快速幂或矩阵快速幂做，首先写出递推式：
设S_i为第i块矿挖出后A的股权，则递推式为
$$\begin{array}{rl}{S}_i& =S_{i-1}+\frac{S_{i-1}}{1+W(i-1)}W\\ & =S_{i-1}\frac{1+Wi}{1+W(i-1)}\\ & =\frac{1+Wi}{1+W(i-1)}\times \frac{1+W(i-1)}{1+W(i-2)}\times \frac{1+Wi-2}{1+W(i-3)}\dots \times S_0(S_0=a)\\ & =(1+Wi)a\end{array}$$
答案就是
$$\begin{array}{rl}\frac{S_i-a}{W}& =ai\\ ans& =na=n\frac{x}{y}\end{array}$$
求出y的逆元相乘取模即可

代码实现

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const ll mod=998244353;
ll n,x,w,y;
ll powmod(ll x,ll p)
{
    ll res=1;
    while(p)
    {
        p&1?res=res*x%mod:0;
        x=x*x%mod;
        p>>=1;
    }
    return res;
}
int main()
{
    cin>>n>>w>>x>>y;
    cout<<n*x%mod*powmod(y,mod-2)%mod;
    return 0;
}