Candy UVA

来源:互联网 发布:淘宝海外代购是真的吗 编辑:程序博客网 时间:2024/06/11 00:24

一个盒子为空,那么这个为空的盒子肯定被取了n次,同时假设另外一个盒子当中还剩余i颗糖果,那么由于另外一个盒子之前也是一共有n颗糖果的,所以这个盒子也被取了n-i次。所以期望就很容易算出来了,但是由于数据可能会很极端,也就是极端的大或者极端的小,所以在解题的时候就使用了对数然后在求解最终的结果的时候取指数这样的一个技巧,具体实现见如下代码:

#include<iostream>#include<vector>#include<string>#include<set>#include<stack>#include<queue>#include<map>#include<algorithm>#include<cmath>#include<iomanip>#include<cstring>#include<sstream>#include<cstdio>#include<deque>#include<functional>using namespace std;const int maxn = 200005;long double logR[maxn];int n;double p;void Init(){logR[0] = 0;for (int i = 1; i <= 200000; i++){logR[i] = logR[i - 1] + log(i);}}long double solve(){long double res = 0;for (int i = 1; i <= n; i++){long double cof = logR[2 * n - i] - logR[n] - logR[n-i];long double v1 = cof + (n + 1)*log(p) + (n - i)*log(1 - p);long double v2 = cof + (n + 1)*log(1 - p) + (n - i)*log(p);res += i*(exp(v1) + exp(v2));}return res;}int main(){Init();int Case = 0;while (cin >> n >> p){Case++;cout << "Case " << Case << ": " <<fixed<<setprecision(6)<< solve() << endl;}return 0;}