Uva1639糖果(大组合数)

题意:

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


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值