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;
}