题意:有两个盒子各有n(n<=2e5)个糖,每天随机选一个(概率分别为p,1-p),然后吃一颗糖,直到有一天,打开盒子一看,没糖了!输入n,p,求此时另一个盒子里糖的个数的数学期望。
思路:根据期望的定义,不妨设最后打开第一个盒子,此时第二个盒子有i颗,则这之前打开过n + (n - i)次盒子,其中有n次取得是盒子1,其余n - i次取得盒子2,概率为C(2 * n - i,n)* p的n + 1次方 * (1 - p)的(n - i)次方。p的指数是n + 1,因为除了前面打开过n次盒子1外,最后又打开了1次。
这个概率表达式在数学上是正确的,但是用计算机计算时需要小心:n可能高达20万,因此C(2 * n - i,n)可能非常大,而p的n + 1次方和(1 - p)的(n - i)次方却非常接近0,。如果分别计算这三项再乘起来,会损失很多精度。一种处理方式是利用对数。。。见紫书333页
#include <bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define INF 0x3f3f3f3f
#define mod 1000000007;
using namespace std;
// freopen("in.txt","r",stdin);
// freopen("out.txt","w",stdout);
const int maxn = 2e5 + 2;
int n;
double p;
long double logc[maxn * 2 + 2];
long double C(int a,int b)
{
return logc[a] - logc[b] - logc[a - b];
}
int main()
{
int kase = 1;
for(int i = 1;i <= maxn * 2;i++)
logc[i] = logc[i - 1] + log(i);
while(scanf("%d%lf",&n,&p) != EOF)
{
double ans = 0;
for(int i = 1;i <= n;i++)
{
long double x = C(2 * n - i,n);
long double v1 = x + (n + 1) * log(p) + (n - i) * log(1 - p);
long double v2 = x + (n + 1) * log(1 - p) + (n - i) * log(p);
ans += (double)i * (exp(v1) + exp(v2));
}
printf("Case %d: %.6f\n",kase++,ans);
}
return 0;
}