BZOJ 1426 收集邮票

Description

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

Input

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

Output

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

Sample Input

3

Sample Output

21.25

HINT

Source

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

概率DP~

设f[i]表示已有i种，集齐n种还需的购买次数的期望值；g[i]表示还需钱数的期望值。显然我们知道f[n]=g[n]=0，要求f[0]和g[0]，所以我们倒着DP。

则f[i]=f[i]*(i/n)+f[i+1]*((n-i)/n)+1，即从现在一张重复和没有重复两种状态更新而来，化简得：

f[i]=f[i+1]+n/(n-i)。

已知f[i]，我们就可以列出g[i]的关系式：g[i]=(g[i+1]+f[i+1])*((n-i)/n)+(g[i]+f[i])*(i/n)+1，化简得：
g[i]=g[i+1]+f[i+1]+(i/(n-i))*f[i]+(n/(n-i))。

然后根据这个DP就好了~

注意所有除法都要加括号并强制转化！

#include<cstdio>
#include<iostream>
using namespace std;

int n;
double f[10001],g[10001];

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