花费期望 King Arthur's Birthday Celebration(poj3682)

题目：
国王庆祝生日，第$i天花费2i-1$（千）个硬币，至于哪一天结束，给定一个正整数$k$，每天抛一次硬币，如果出现第$k$次正面朝上，就结束庆祝，正面朝上的概率为$p$。
问庆祝的天数和花费的金币的期望。

思路：

设$f[i]为第i$天结束的概率，则$f(i)=\binom{i-1}{k-1}\times p^k\times (1-p)^{i-k}$，总概率为 $\sum f(i)=1$，其中$i∈[k，+∞）$，以下均为此范围
则有，$p^k*\sum_{i=k}^\infty\left(\binom{i-1}{k-1}*(1-p)^{i-k}\right)=1$，即：$∑_{i=k}^\infty\left(\binom{i-1}{k-1}*(1-p)^{i-k}\right)=\frac 1{p^k}$，
令 $g(k)=∑_{i=k}^\infty\left(\binom{i-1}{k-1}*(1-p)^{i-k}\right)=\frac 1{p^k}$
那么，天数的期望：

$$\begin{align} days &=∑_{i=k}^\infty(i*f(i))\\ &=p^k\times∑_{i=k}^\infty\left(i\times\binom{i-1}{k-1}\times(1-p)^{i-k}\right)\\ &=kp^k\times∑_{i=k}^\infty\left(\binom ik\times(1-p)^{i-k}\right)\\ &=kp^kg(k+1)\\ &=k\times\frac{p^k}{p^{k+1}}\\ &=\frac kp\\ \end{align}$$

花费的期望：

$$\begin{align} cost&=∑_{i=k}^\infty(i^2 f(i)) //第i天结束的话就花费1+3+5+...+2i-1=i^2;\\ &=p^k\times∑\left(i^2\binom{i-1}{k-1}\times(1-p)^{i-k}\right)\\ &=kp^k\times∑\left(i\binom ik\times(1-p)^{i-k}\right)\\ &=kp^k\times∑\left((i+1)\times\binom ik\times(1-p)^{i-k}\right)-kp^k\times∑\left(\binom ik*(1-p)^{i-k}\right)\\ &=k(k+1)p^k∑\left(\binom{i+1}{k+1}*(1-p)^{i-k}\right)-kp^kg(k+1)\\ &=k(k+1)\times\frac{p^k}{p^{k+2}}-days\\ &=\frac{k(k+1)}{p^2}-days\\ &=days\times\frac{k+1}p-days \end{align}$$

#include <iostream>
#include <cmath>
#include <stdio.h>
#include <map>
#include <algorithm>
using namespace std;
#define LL long long 
#define MAX 1000010

int main()
{
    int n,k;
    double p;
    while (scanf("%d",&k),k)
    {
        scanf("%lf",&p);
        double ans=k/p;
        printf("%.3lf %.3lf\n",ans,ans*(k+1)/p-ans);
    }

}