單選錯位

Time Limit: 10 Sec Memory Limit: 259 MB

Description

Input

n很大，為了避免讀入耗時太多，
輸入文件只有5個整數參數n, A, B, C, a1，
由上交的程序產生數列a。
下面給出pascal/C/C++的讀入語句和產生序列的語句（默認從標準輸入讀入）：
// for pascal
readln(n,A,B,C,q[1]);
for i:=2 to n do q[i] := (int64(q[i-1]) * A + B) mod 100000001;
for i:=1 to n do q[i] := q[i] mod C + 1;
// for C/C++

Output

輸出一個實數，表示gx期望做對的題目個數，保留三位小數。

Sample Input

3 2 0 4 1

Sample Output

1.167

【樣例說明】

a[] = {2,3,1}

正確答案 gx的答案 做對題目 出現概率

{1,1,1} {1,1,1} 3 1/6

{1,2,1} {1,1,2} 1 1/6

{1,3,1} {1,1,3} 1 1/6

{2,1,1} {1,2,1} 1 1/6

{2,2,1} {1,2,2} 1 1/6

{2,3,1} {1,2,3} 0 1/6

共有6種情況，每種情況出現的概率是1/6，gx期望做對(3+1+1+1+1+0)/6 = 7/6題。（相比之下，lc隨機就能期望做對11/6題）

我從來都不水博客的。。。。我們由這道題可以學會期望可以相加。。。。
\(ans=\sum(每道題做對的期望)\)

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e7 + 5;
int n, A, B, C, a1;
double ans = 0;
int a[maxn];

inline void putit()
{
    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; 
}

inline void workk()
{
    for(int i = 1; i < n; ++i)  ans += (1.0) / max(a[i], a[i + 1]);
    ans += (1.0) / max(a[n], a[1]);
}

int main()
{
    putit();
    workk();
    printf("%.3f\n", ans);
    return 0;
}

bzoj2134 單選錯位