BZOJ2134

2134: 单选错位

Time Limit: 10 Sec   Memory Limit: 259 MB
Submit: 340   Solved: 251
[ Submit][ Status]

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++ 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; 选手可以通过以上的程序语句得到n和数列a（a的元素类型是32位整数），n和a的含义见题目描述。

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题）
【数据范围】
对于100%的数据 2≤n≤10000000, 0≤A,B,C,a1≤100000000

【题解】简单的一道概率题。关键在于分析如何计算它的情况。

           我们考虑到要答对一道题，必须满足 i-1 的答案等于 i 的答案，所以只要分析 p [ i ] == p[ i-1 ]的概率问题。这里我们可以分类讨论：

  1、当 a [ i -1 ] >= a [ i ] , 那么使答案在a [ i  ]范围内的概率为 a[ i ] / a[ i -1 ] ,然后乘上 答对的 概率 1 /a[ i ],

就是 两者相等的概率 即 a[ i ] / a[ i - 1]*1/ a[ i ] ==1 / a[ i -1];

                2、当 a[ i -1]<=a[ i ],那么同上分析，答对的概率为 1/a[ i ];

所以总的期望答对题的数量为 ∑ 1 / max(a[i-1],a[i]);最后在判断一下a[n]和a[1],就行了。

代码中我加了一个自认为有效的优化（精度），但不加优化交上去也过了。

//加优化；

#include<cstdio>
#define LL long long
#define mod 100000001

const long long max=(long long)1<<36;
int a[10000001];
int main()
{
	int A,n,B,C;
	LL h,t;double ans=0;
	scanf("%d%d%d%d%d",&n,&A,&B,&C,&a[1]);
	for(int i=2;i<=n;i++)a[i]=((LL)a[i-1]*A+B)%mod;
	h=0;t=1;a[1]=a[1]%C+1;
	for(int i=2;i<=n;i++)
	{
		a[i]=a[i]%C+1;
		if(a[i-1]>=a[i])
		{
			h=h*a[i-1]+t;t=t*a[i-1];
		}else
		{
			h=h*a[i]+t;t=t*a[i];
		}
		if((h>max)||(t>max))
			{
				ans+=(double)h/t;
				h=0;t=1;
			}
	}
	if(a[n]>=a[1]){
		h=h*a[n]+t;t=t*a[n];
	}else {h=h*a[1]+t;t=t*a[1];}
	ans+=(double)h/t;
	printf("%.3lf\n",ans);
	return 0;
}

//不加优化；

#include<cstdio>
#define LL long long
#define mod 100000001
 
int max(int x,int y){return x>y?x:y;}
int a[10000001];
int main()
{
    int A,n,B,C;
    double ans=0;
    scanf("%d%d%d%d%d",&n,&A,&B,&C,&a[1]);
    for(int i=2;i<=n;i++)a[i]=((LL)a[i-1]*A+B)%mod;
    a[1]=a[1]%C+1;
    for(int i=2;i<=n;i++)
    {
        a[i]=a[i]%C+1;
        ans+=(double)1.0/max(a[i-1],a[i]);
    }
    ans+=(double)1.0/max(a[n],a[1]);
    printf("%.3lf\n",ans);
    return 0;
}