【BZOJ】2134: 单选错位 期望DP

【题意】有n道题，第i道题有ai个选项。把第i道题的正确答案填到第i+1道题上（n填到1），问期望做对几道题。n<=10^7。

【算法】期望DP

【题解】正确答案的随机分布不受某道题填到后面是否正确影响，因此每道题对的期望都是独立的。

从排列的角度分析，对每道题有a[i-1]个选择和a[i]个选项，共a[i-1]*a[i]种排列，其中只有min(a[i-1],ai)种排列使这道题正确，所以

$$E(i)=\frac{Min(a[i-1],a[i])}{a[i-1]*a[i]}=\frac{1}{Max(a[i-1],a[i])}$$

然后根据期望的线性相加。

复杂度O(n)。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=;
int n,a[maxn];
int main()
{
    int A,B,C;
    scanf("%d%d%d%d%d",&n,&A,&B,&C,&a[]);
    for (int i=;i<=n;i++) a[i] = ((long long)a[i-] * A + B) % ;
    for (int i=;i<=n;i++) a[i] = a[i] % C + ;
    a[]=a[n];
    double ans=;
    for(int i=;i<=n;i++)ans+=1.0/max(a[i],a[i-]);
    printf("%.3lf",ans);
    return ;
} 

如果实在纠结前面题对和后面题对有一题重合，考虑期望可以线性相加，所以实际上是可以拆出来计算的。