P4550 数学期望 + DP

题意

传送门 P4550 收集邮票

题解

随机变量 $x_i$ 为已取得 $i$ 种邮票的状态，到取得所有种类邮票的状态，还需要取邮票的次数。则最终需要花费 $\sum _{i=1}^{x_0}i=x_0(x_0+1)/2$ 元。需要求解 $E(x_i),E(x_i^2)$，分别设为 $f(i),g(i)$。考虑从当前状态购买一张邮票的状态转移，根据递归的思路得到 $$E(x_i)=\frac{i}{n}E(x_i+1)+\frac{n-i}{n}E(x_{i+1}+1)$$ $$E(x_i^2)=\frac{i}{n}E[(x_i+1{)}^2]+\frac{n-i}{n}E[(x_{i+1}+1{)}^2]$$ 即 $$f(i)=f(i+1)+\frac{n}{n-i}$$ $$g(i)=\frac{2i}{n-i}f(i)+\frac{n}{n-i}+g(i+1)+2f(i+1)$$

#include <bits/stdc++.h>
using namespace std;
const int maxn = 10005;
int N;
double f[maxn], g[maxn];

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