BZOJ.2134.[国家集训队]单选错位(概率 递推)

时间:2023-03-08 16:27:02

题目链接

如题目中的公式,我们只要把做对每个题的概率加起来就可以了(乘个1就是期望)。

做对第i道题的概率 $$P_i=\frac{1}{max(a_{i-1},a_i)}$$

原式是 \(P_i=\frac{min(a_{i-1},a_i)}{a_{i-1}\times a_i}\),化简后得到上式。

例:假设第i-1道有3个选项,第i道有5个选项,暴力一点,那么做对就是从3个中选1个和从5个中选1个相同的概率,

概率为 \(\frac{1}{3}\times\frac{1}{5}+\frac{1}{3}\times\frac{1}{5}+\frac{1}{3}\times\frac{1}{5}=\frac{min(3,5)}{3\times 5}\)

//820kb	836ms
#include <cstdio>
#include <algorithm>
#define mod (100000001) int n,A,B,C,a_1; int main()
{
scanf("%d%d%d%d%d",&n,&A,&B,&C,&a_1);
double res=0;
int pre=a_1, a_pre=a_1%C+1, now, a_now;
for(int i=2; i<=n; ++i)
{
now=(1ll*pre*A+B)%mod, a_now=now%C+1;//直接longlong略慢
res+=1.0/std::max(a_pre,a_now);
pre=now, a_pre=a_now;
}
res+=1.0/std::max(a_1%C+1,a_now);
printf("%.3lf",res); return 0;
}