概率论——赠券收集问题

概率论——赠券收集问题

题意

有$n$中不同的邮票。现在想购买所有种类的邮票，第$i$次购买一张邮票需要花费$i$元，并且购买到的这张邮票的种类的概率均等即均为$\frac{1}{n}$，求花费的期望。

题解

根据等差数列求和公式，购买次数为$x$的花费为$\frac{x(x+1)}{2}$，我们所求的就是$E(\frac{x(x+1)}{2})$，即：

$$E(\frac{x(x+1)}{2})=E(\frac{x^2+x}{2})=\frac{E(x^2)+E(x)}{2}$$

根据方差与期望之间的关系$E(x^2)=D(x)+E^2(x)$，我们有：

$$E(\frac{x(x+1)}{2})=\frac{D(x)+E^2(x)+E(x)}{2}$$

此问题就变成了：

有$n$中不同的邮票。现在想购买所有种类的邮票，一次只能购买一张邮票，并且购买到的这张邮票的种类的概率均等即均为$\frac{1}{n}$，求购买的次数的期望和方差。

我们定义$x_i$为从已经买到了$i-1$张不同种类的邮票开始，到购买到$i$张不同种类的邮票结束所购买的次数，根据期望的线性性，有：

$$E(x)=E(\sum _{i=1}^n{x}_i)=\sum _{i=1}^{n}E(x_i)$$

其中$x_i\sim G(\frac{n-i+1}{n})$，$G$为几何分布。

几何分布$x\sim G(p)$，期望$E(x)=\frac{1}{p}$，方差$D(x)=\frac{1-p}{p^2}$

$$E(x_i)=\frac{n}{n-i+1}$$

则带入上式：

$$E(x)=\sum _{i=1}^n\frac{n}{n-i+1}=n\sum _{i=1}^n\frac{1}{i}=n{H}_n$$

下一步，求$D(x)$，因为$x_i$之间相互独立，协方差等于$0$，故：

$$\begin{gathered} D(x)=D(\sum _{i=1}^n{x}_i)=\sum _{i=1}^{n}D(x_i) \\ =\sum _{i=1}^n\frac{n(i-1)}{(n-i+1{)}^2}=n\sum _{i=1}^n\frac{n-(n-i+1)}{(n-i+1{)}^2} \\ =n^2\sum _{i=1}^n\frac{1}{i^2}-n\sum _{i=1}^n\frac{1}{i} \\ =n^2{H}_n^{(2)}-n{H}_n \end{gathered}$$

带入上式：

$$\begin{gathered} E(\frac{x(x+1)}{2})=\frac{D(x)+E^2(x)+E(x)}{2} \\ =\frac{n^2(H_n^{(2)}+H_n^2)}{2} \end{gathered}$$

代码

我们只需要循环计算一下调和数列的值即可，时间复杂度$O(n)$，空间复杂度$O(1)$。

int main()
{
    int n;
    scanf("%d", &n);
    double h = 0;
    double h2 = 0;
    for (int i = 1; i <= n; i++)
    {
        h += 1.0 / i;
        h2 += (1.0 / i) * (1.0 / i);
    }
    double ans = (h2 + h * h) * n * n / 2;
    printf("%.2lf", ans);
    return 0;
}

总结

• 期望的线性性（如何对期望进行拆解）
• 期望与方差的关系
• 几何分布的期望和方差的公式