hdu 4465 Candy

本文介绍了一个关于两个装有相同数量糖果的盒子的概率问题。通过递推公式计算在一个盒子为空时,另一个盒子中剩余糖果数量的期望值。采用对数处理避免数值计算中的溢出问题,并使用预计算对数表提高效率。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

        题意:有两个盒子,里面各有n颗糖果。每天以p的概率打开第一个盒子,1-p的概率打开另一个盒子,拿一颗糖吃掉。求当你打开发现是空盒的时候,另一个盒子里的糖的期望。

        思路:递推。计算一个盒子空的时候,另一个盒子还有i颗糖的概率,把它乘上i并累加起来。公式是f(i)=p^n*(1-p)^i*C(i+n,i)。计算时肯定不能暴力,算出来i=0,就可以递推i=1,2,3...,f(i)=f(i-1)*(1-p)*i+n/i。

        然后你会发现,中间结果太小,爆了double,取对数解决。然后还是不行,超时,打个1~400000的对数表,就过了。。。


#include<iostream>
#include<cmath>
#include<queue>
#include<map>
#include<set>
#include<vector>
#include<algorithm>
#include<string.h>
#include<cstdio>

using namespace std;

const double E=2.718281828459045;
double tab[400010];

int main(){
	for(int i=1;i<=400000;i++){
		tab[i]=log(i);
	}
	
	int n;
	double p;
	int cas=0;
	while(cin>>n>>p){
		cas++;
		double logx=n*log(p);
		int N=n+1;
		int K=1;
		double ans1=n*pow(E,logx);
		double log1_p=log(1-p);
		for(int i=1;i<n;i++){
			
			logx+=tab[N];
			logx-=tab[K];
	
			logx+=log1_p;
			N++;
			K++;
			
			ans1+=(n-i)*pow(E,logx);
			
		}
		ans1*=(p);

		p=1-p;
		logx=n*log(p);
		N=n+1;
		K=1;
		double ans2=n*pow(E,logx);
		log1_p=log(1-p);
		for(int i=1;i<n;i++){
			logx+=tab[N];
			logx-=tab[K];
			logx+=log1_p;
			N++;
			K++;
			ans2+=(n-i)*pow(E,logx);
		}
		ans2*=(p);
		printf("Case %d: %.6lf\n",cas,ans1+ans2);
	}
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值