BZOJ1426收集邮票

题目描述

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

---

输入

一行,一个数字N, $N$<=$10000$

---

输出

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

---

Solution

关于数学期望比较有意思的一道题
难点在于每次价格不一样
首先想一想如果价格一样怎么做,就是每张邮票价格都为1怎么办?
设$f[i]$表示我已经买了i张邮票,还要花的钱
显然$f[n]=0$,求$f[0]$
比较容易得到

$f[i]=\frac{i}{n}*f[i]+\frac{n-i}{n}*f[i+1]+1$

有$\frac{i}{n}$的概率选到前i种已经选过的
还有$\frac{n-i}{n}$选到没选过的

然后移项变成

$f[i]=f[i+1]+\frac{n}{n-i}$

因为每张价格都是1,所以也可以等价成数量
现在要求每张的价格都是上一张的价格+1
设$g[i]$是已经买了i张邮票,还要花的钱,
因为每次买的价格都是上一张的价格+1

$g[i]=\frac{i}{n}$$*$$(g[i]+f[i])$$+$$\frac{n-i}{n}*(g[i+1]+f[i+1])$$+1$

移项得到

$g[i]=g[i+1]+f[i+1]+\frac{i*f[i]+n}{n-i}$

逆推得到$g[0]$就是答案

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 100010
double f[maxn],g[maxn];
int main(){
    double n;scanf("%lf",&n);
    for(int i=n-1;~i;--i) f[i]=f[i+1]+n/(n-i);
    for(int i=n-1;~i;--i) g[i]=g[i+1]+f[i+1]+(f[i]*i+n)/(n-i);
    printf("%.2lf",g[0]);
    return 0;
}