题意:
有两个盒子各有n个糖果(n<=200000),每天随机选择一个:选第一个盒子的概率是p(0 ≤ p ≤ 1),第二个盒子的概率为1-p,然后吃掉其中的一颗。直到有一天,随机选择一个盒子打开一看,没糖了!现在请你计算另一个盒子里剩下的糖果数量的期望值。
思路:
就是期望,没什么说的。
我们假设到第n天的时候取得是第1个盒子的糖,此时第2个盒子有i颗糖,则在此之前打开了n+(n-i)次盒子, 其中n次打开了第一个盒子,(n-i)次打开了第二个盒子,则概率是C(2n-i,n)*p(n+1)*(1-p)n-i。
由于n高达20w,所以二次项系数会非常大,而后面的概率会非常小,所以如果直接计算会爆精度,所以这里我们用求对数的方法进行计算。
log(i) 以e为底;exp(i),求e^i
// 对数和指数函数
#include <cstdio>
#include <cmath>
#include <algorithm>
#define fi first
#define se second
#define pii pair<int,int>
using namespace std;
typedef long long LL;
const int maxn = 200000+1;
long double logFac[maxn];
long double logC(int n, int m){
return logFac[n] - logFac[m] - logFac[n-m];
}
double solve(int n, double p){
double ans = 0;
double t1 = (double)(n+1)*log(p);
double t2 = (double)(n+1)*log(1-p);
for(int i = 0; i <= n; ++i){
double c = logC(2*n-i, n);
double v1 = c + t1 + (double)(n-i)*log(1-p);
double v2 = c + t2 + (double)(n-i)*log(p);
ans += i * (exp(v1) + exp(v2));
}
return ans;
}
int main()
{
freopen("in.txt","r",stdin);
int kase = 1;
logFac[0] = 0;
for(int i = 1; i < maxn; ++i){
logFac[i] = logFac[i-1] + log(i);
}
int n;
double p;
while(scanf("%d%lf",&n,&p) == 2&&n){
double ans = solve(n,p);
printf("Case %d: %lf\n",kase++,ans);
}
return 0;
}