I题: Incentive Model
原题链接:https://ac.nowcoder.com/acm/contest/11260/I
题目大意
有A、B两人一起逐个挖 n ( 1 ≤ n < 998244353 ) n(1\le n<998244353) n(1≤n<998244353) 块矿,对于同一块矿,有 a a + b \frac{a}{a+b} a+ba 的概率属于A,有 b a + b \frac{b}{a+b} a+bb 的概率属于B,其中 a , b a,b a,b 为双方当时的股权占比,初始 a + b = 1 a+b=1 a+b=1 。某方拥有一块矿后,其股权增加 w w w 。初始A拥有 x y \frac{x}{y} yx 的股权,B拥有 1 − x y 1-\frac{x}{y} 1−yx 的股权。求A能获得块的期望数。
题解
我们设
S
i
S_i
Si 表示挖了第
i
i
i 块矿后A的期望股权,
易得递推式:
S
i
=
S
i
−
1
+
S
i
−
1
1
+
(
i
−
1
)
w
w
S_i=S_{i-1}+\frac{S_{i-1}}{1+(i-1)w}w
Si=Si−1+1+(i−1)wSi−1w
S
i
−
1
S_{i-1}
Si−1 表示上一块时的股权,
1
+
(
i
−
1
)
w
1+(i-1)w
1+(i−1)w 表示挖了
i
−
1
i-1
i−1 块后AB两人的总股权,
S
i
−
1
1
+
(
i
−
1
)
w
\frac{S_{i-1}}{1+(i-1)w}
1+(i−1)wSi−1 为该块属于A的概率,乘以
w
w
w 即是期望增加的股权。
n
n
n 的数据范围很大,直接递推显然会TLE,我们考虑将递推式转化为通项式,进行化简如下:
S
i
=
S
i
−
1
(
1
+
w
1
+
(
i
−
1
)
w
)
S
i
=
S
i
−
1
1
+
i
⋅
w
1
+
(
i
−
1
)
w
\begin{aligned} S_i&=S_{i-1}(1+\frac{w}{1+(i-1)w})\\ S_i&=S_{i-1}\frac{1+i·w}{1+(i-1)w}\\ \end{aligned}
SiSi=Si−1(1+1+(i−1)ww)=Si−11+(i−1)w1+i⋅w
我们将式子展开:
S
i
=
1
+
i
⋅
w
1
+
(
i
−
1
)
w
⋅
1
+
(
i
−
1
)
w
1
+
(
i
−
2
)
w
⋅
1
+
(
i
−
2
)
w
1
+
(
i
−
3
)
w
.
.
.
1
+
w
1
⋅
S
0
S_i=\frac{1+i·w}{1+(i-1)w}·\frac{1+(i-1)w}{1+(i-2)w}·\frac{1+(i-2)w}{1+(i-3)w}...\frac{1+w}{1}·S_0
Si=1+(i−1)w1+i⋅w⋅1+(i−2)w1+(i−1)w⋅1+(i−3)w1+(i−2)w...11+w⋅S0
约分后得(
S
0
S_0
S0 即初始的
a
a
+
b
=
a
(
a
+
b
=
1
)
\frac{a}{a+b}=a(a+b=1)
a+ba=a(a+b=1) ):
S
i
=
a
(
1
+
i
⋅
w
)
S_i=a(1+i·w)
Si=a(1+i⋅w)
此时期望获得的块数为
S
i
−
a
w
=
a
⋅
i
⋅
w
+
a
−
a
w
=
a
⋅
i
\frac{S_i-a}{w}=\frac{a·i·w+a-a}{w}=a·i
wSi−a=wa⋅i⋅w+a−a=a⋅i 。
可得答案
a
n
s
=
n
⋅
a
=
n
x
y
ans=n·a=n\frac{x}{y}
ans=n⋅a=nyx
参考代码
#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;
}