hdu 4465 求期望（C（m,n）太大用log优化）

/*
坑啊
数学函数的运用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;}

转载于:https://www.cnblogs.com/thefirstfeeling/p/4410558.html