[水题日常]UVA1639 糖果(Candy,ACM/ICPC Chengdu 2012)
- 今天来尝试了几道数学期望相关的题,这是我认为比较有趣的一道题
- 这次不废话啦直接开始~
- 一句话题意:两个分别装有n个糖果的盒子,每次随机选一个盒子然后拿走一颗糖(选的概率分别是\(p\)和\((1-p)\)),直到有一次选了一个盒子打开之后发现没有糖果了,求另一个盒子糖果个数的期望值
- \(n<=2*10^5\)
嗯首先我们得知道期望的线性性质:\(E(X+Y)=E(X)+E(Y)\)
-
如果我们假设最后打开的是第一个盒子,那么第二个盒子的糖果个数其实可能有\(0~n\)个,如果已经知道了有\(i\)个的概率就好了,这样根据期望的线性性质答案就是把所有的\(i*P(i)\)加起来(当然还要考虑最后打开第二个盒子)
-
好的我们现在来具体考虑第二个盒子有\(i\)个糖果的情况,第二个盒子还有\(i\)个的话那么在这之前盒子一定被打开过了\(n+n-i=2n-i\)次(\(n\)次取第一个盒子,\(n-i\)次取第二个),这样一来概率就是\(C(2n-i,n)p^{n+1}(1-p)^{n-i}\)(因为在这之前取了\(n\)次第一个盒子,这次又取了一次)
-
然后每次枚举一下\(i\)对两种情况(最后打开第一个或第二个盒子)求个和是不是就可以了?
-
蓝鹅这样子在理论上是正确的,但是实际求的时候\(C(2n-i,n)\)可以挺大的,而\(p^{n+1}\)和\((1-p)^{n-i}\)会很小,直接算的话会有精度误差(当然如果你要写高精度我也不拦你~),而且这个误差应该会比较大,大概超出了题目的要求(我之前改进后过程用double好像都wa掉了…),那么怎么办呢?
-
紫书上关于这题介绍了一种取对数的处理方法,具体来说就是在算的过程中全部取对数(我这里是\(e\)为底啦…不过好像一般也都是这样),比如最后打开第一个盒子时第二个盒子有\(i\)个糖果的概率取\(e\)的对数就是\(v1(i)=ln(C(2n-i,n))+(n+1)ln(p)+(n-i)ln(1-p)\)啦(第二个盒子的情况同理),最后把答案加上\(i(e^{v1(i)}+e^{v2(i)})\)就好啦~
-
预处理阶乘的对数的时候注意可能会用到\(n\)的两倍
-
再具体的话就没什么好说的啦
-
还是存一下代码吧~
#include<cstdio> #include<cmath> typedef long double ldouble; const int N=200005; ldouble p,LogF[N<<1]; inline double getC(int n,int k) { return LogF[n]-LogF[k]-LogF[n-k]; } inline double solve(int n) { double ans=0.0; for(int i=0;i<=n;i++) { ldouble c=getC(2*n-i,n); ldouble v1=c+log(p)*(n+1)+log(1-p)*(n-i); ldouble v2=c+log(p)*(n-i)+log(1-p)*(n+1); ans+=i*(exp(v1)+exp(v2)); }return ans; } int main() { int n,kase=0;LogF[0]=0; for(int i=1;i<(N<<1);i++)LogF[i]=LogF[i-1]+log(i); while(scanf("%d%Lf",&n,&p)==2) printf("Case %d: %.6lf\n",++kase,solve(n)); return 0; }
如果有错欢迎指出~
撒椛~(雾)