洛谷P1297 [国家集训队]单选错位_数学期望

考虑第 $i$ 位， 那么当前共有 $a[i]$ 种选项，那么当前选项正确的情况就是下一个被误填的答案与当前答案相同。换句话说，当前答案一共有 $a[i]$ 种可能，而下一个答案有 $a[i+1]$种可能，那么总共有 $a[i]\ast a[i+1]$ 种可能，其中，我们要去 $min(a[i],a[i+1])$ 作为分子（想一想，为什么）。故每种答案的贡献为 $1/max(a[i],a[i+1])$，最后加和即可。
时间复杂度为 $O(n)$
Code：

#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn=10000000+3;
int a[maxn];
double get(int a,int b){
	double up=1.0;
	double down=max(a,b);
	return (double)up/down;
}
int main()
{
    int n,A,B,C;
    scanf("%d%d%d%d%d",&n,&A,&B,&C,a+1);
    for (int i=2;i<=n;i++)
        a[i] = ((long long)a[i-1] * A + B) % 100000001;
    for (int i=1;i<=n;i++) a[i] = a[i] % C + 1;
    double ans=0.0;
    for(int i= 1;i< n;++i)
        ans+=get(a[i],a[i+1]);
	ans+=get(a[1],a[n]);
    printf("%.3f",ans);
    return 0;
}