Candy UVA - 1639

题意:有两个盒子各有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;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值