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

I题: Incentive Model

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

题目大意

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

题解

我们设 $S_i$ 表示挖了第 $i$ 块矿后A的期望股权，
易得递推式:
$$S_i=S_{i-1}+\frac{S_{i-1}}{1+(i-1)w}w$$

$S_{i-1}$ 表示上一块时的股权， $1+(i-1)w$ 表示挖了 $i-1$ 块后AB两人的总股权， $\frac{S_{i-1}}{1+(i-1)w}$ 为该块属于A的概率，乘以 $w$ 即是期望增加的股权。
$n$ 的数据范围很大，直接递推显然会TLE，我们考虑将递推式转化为通项式，进行化简如下:
$$\begin{array}{rl}{S}_i& =S_{i-1}(1+\frac{w}{1+(i-1)w})\\ S_i& =S_{i-1}\frac{1+i\cdot w}{1+(i-1)w}\end{array}$$

我们将式子展开:
$$S_i=\frac{1+i\cdot w}{1+(i-1)w}\cdot \frac{1+(i-1)w}{1+(i-2)w}\cdot \frac{1+(i-2)w}{1+(i-3)w}...\frac{1+w}{1}\cdot S_0$$

约分后得( $S_0$ 即初始的 $\frac{a}{a+b}=a(a+b=1)$ ):
$$S_i=a(1+i\cdot w)$$

此时期望获得的块数为 $\frac{S_i-a}{w}=\frac{a\cdot i\cdot w+a-a}{w}=a\cdot i$ 。
可得答案 $ans=n\cdot a=n\frac{x}{y}$

参考代码

#include<bits/stdc++.h>
#define mod 998244353
using namespace std;

int inv(int x){//逆元
	int ret=1,p=mod-2;
	for(;p;x=1ll*x*x%mod,p>>=1)if(p&1)ret=1ll*ret*x%mod;
	return ret;
}

int main()
{
	int n,w,x,y;
	cin>>n>>w>>x>>y;
	cout<<1ll*n*x%mod*inv(y)%mod<<endl;//注意取模
	return 0;
}