您的位置:首页 > 其它

hdu 4465 求期望(C(m,n)太大用log优化)

2014-11-04 12:33 197 查看
/*
坑啊
数学函数的运用log处理,exp还原
tle好长时间,一直用g++交,最后把别人正确的代码交上也是tle,用c++交一遍ac
题意:有两个数量为n的糖果,一个人开始吃,吃到最后有一堆剩余为0的时候不吃,把另一堆留给另外一个另一个人,求另一个人得到糖果的期望,
这个人吃第一堆糖果的概率是p,第二堆糖果的概率是1-p,并且当他把一堆糖果吃完,再次吃的时候发现没有了才结束。
解:分别求出当第一堆剩余的期望,第二堆剩余的期望加起来
(n-i)*C(n+i,i)*pow(p,n+1)*pow(1-p,i)+(n-i)*C(n+i,i)*pow(1-p,n+1)*pow(p,i);
*/
#include<stdio.h>
#include<string.h>
#include<math.h>
double ee;
double  cc(double e,int k,double p,int n,double q,int m) {
double sum;
sum=ee=e+log(1.0*k)-log(1.0*(k-n+1));
sum+=1.0*n*log(p);
sum+=1.0*m*log(q);
return sum;
}
int main() {
int n,i,k=0;
double p,e,sum;
while(scanf("%d%lf",&n,&p)!=EOF) {
ee=0;e=0;
sum=1.0*n*pow(p,n+1)+1.0*n*pow(1-p,n+1);
for(i=1;i<=n;i++) {
e=ee;
sum+=1.0*(n-i)*(exp(cc(e,n+i,p,n+1,1-p,i))+exp(cc(e,n+i,1-p,n+1,p,i)));
}
printf("Case %d: %.6f\n",++k,sum);
}
return 0;}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: