1. 程式人生 > >洛谷P1297 [國家集訓隊]單選錯位_數學期望

洛谷P1297 [國家集訓隊]單選錯位_數學期望

考慮第 ii 位, 那麼當前共有 a[i]a[i] 種選項,那麼當前選項正確的情況就是下一個被誤填的答案與當前答案相同。換句話說,當前答案一共有 a[i]a[i] 種可能,而下一個答案有 a[i+1]a[i + 1]種可能,那麼總共有 a[i]a[i+1]a[i]*a[i+1] 種可能,其中,我們要去 min(a[i],a[i+1])min(a[i], a[i+1]) 作為分子(想一想,為什麼)。故每種答案的貢獻為 1/max(a[i],a[i+1])1/max(a[i],a[i+1])

+1]),最後加和即可。 時間複雜度為 O(n)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; }