[bzoj1426] 收集邮票（概率dp）

Description

有$n$种不同的邮票，皮皮想收集所有种类的邮票。唯一的收集方法是到同学凡凡那里购买，每次只能买一张，并且买到的邮票究竟是$n$种邮票中的哪一种是等概率的，概率均为$1/n$。但是由于凡凡也很喜欢邮票，所以皮皮购买第$k$张邮票需要支付$k$元钱。现在皮皮手中没有邮票，皮皮想知道自己得到所有种类的邮票需要花费的钱数目的期望。

Input

一行,一个数字N，$N \leq 10000$

Output

要付出多少钱. 保留二位小数

Sample Input

3

Sample Output

21.25

题解

用$f[i]$表示已经拥有了$i$张邮票，则期望还需要购买的邮票数。
则$f[n]=0$,$f[i]=f[i] *\frac{n-i}{n}+f[i+1]*\frac{i}{n}+1$。
整理得$f[i]=f[i+1]+\frac{n}{n-i}$。

再设$g[i]$为已经拥有了$i$张邮票，还需要的钱。
则$g[i]=(g[i+1]+f[i+1])*\frac{n‐i}{n}+(g[i]+f[i])*\frac{i}{n}+1$
至于这个公式怎么推的我也不是很清楚，大概就是因为每次买的时候花的钱都要比上一次多 1 1 <script type="math/tex" id="MathJax-Element-1192">1</script>。

Code

#include<bits/stdc++.h>
using namespace std;

double n,f[10010],g[10010];

int main()
{
    scanf("%lf",&n);
    for(int i=n-1;i>=0;i--)f[i]=f[i+1]+n/(n-i);
    for(int i=n-1;i>=0;i--)g[i]=g[i+1]+f[i+1]+i/(n-i)*f[i]+n/(n-i);
    printf("%.2lf",g[0]);
    return 0;
}