JZOJ3377 【NOI2013模拟】抽奖(lottery) 期望的线性性

本文链接：https://blog.csdn.net/YxuanwKeith/article/details/52624033

题目大意

$N$ 位同学参与抽奖，每位同学在抽奖箱中放入了 $A_i$ 写有自己姓名的小球。抽奖总共有 $M$ 轮，每轮随机地抽出一个小球并记录。之后，将这个小球放回抽奖箱。每位同学如果写有他的名字的球被抽出了 $X$ 次，则他将获得奖金 $X^2$ 元。现在要求计算总奖金的期望值和获得奖金大于0的人数的期望值。

$N,M\leq 10^5$
$\sum A_i \leq 10^9$

解题思路

先设 $p_i$ 表示一个抽到第 $i$ 个人的概率。
我们把每次抽奖分开考虑，假设我们之前已经抽了 $t$ 个球，现在我们设 $t+1$ 个球抽到第 $i$ 个人，这个人已经被抽到 $m_i$ 次。那么这一个球带来的贡献 $\Delta=(m_i+1)^2-m_i^2=2m_i+1$ 。那么我们考虑对两边求期望，根据期望的线性性

E [Δ i] = E [2 m i + 1] = 2 E [m i] + 1 E [m i] = t * p i

$E[\Delta_i]=E[2m_i+1]=2E[m_i]+1\\ E[m_i]=t*p_i$
所以

E[Δi]=2t∗pi+1 $E[\Delta_i]=2t*p_i+1$ 。那么对于一个一次抽奖的期望就很好计算了：

E [Δ] = \sum i = 1 N (p i * E [Δ i]) = \sum i = 1 N (2 t * p 2 i + p i) = 2 t \sum i = 1 N p 2 i + 1

$E[\Delta]=\sum_{i=1}^{N} (p_i*E[\Delta_i])=\sum_{i=1}^{N}(2t*p_i^2+p_i)=2t\sum_{i=1}^{N}p_i^2+1$
那么对于第一问的答案只需把

t=1...N $t=1...N$ 的情况加起来就可以了。

A n s 1 = \sum t = 0 M - 1 (2 t * \sum i = 1 N p 2 i + 1) = 2 (\sum t = 0 M - 1 t) (\sum i = 1 N p 2 i) + M = M (M - 1) \sum i = 1 N p 2 i + M

$Ans_1=\sum_{t=0}^{M-1}(2t*\sum_{i=1}^{N}p_i^2+1)=2(\sum_{t=0}^{M-1}t)(\sum_{i=1}^{N}p_i^2)+M=M(M-1)\sum_{i=1}^{N}p_i^2+M$
同样对于第二问也十分好求，由于期望的线性性，我们考虑对每个人分开考虑然后加起来，显然

A n s 2 = \sum i = 1 N [1 - (1 - p i) M] = N - \sum i = 1 N (1 - p i) M

$Ans_2=\sum_{i=1}^{N}[1-(1-p_i)^M]=N-\sum_{i=1}^{N}(1-p_i)^M$

程序

//YxuanwKeith
#include <cstring>
#include <cstdio>
#include <algorithm>

using namespace std;

const int MAXN = 1e5 + 5;

int N, M, Num[MAXN];
double P[MAXN];

double Power(double x, int y) {
    double Ans = 1;
    for (; y; y >>= 1, x = x * x) 
        if (y & 1) Ans = Ans * x;
    return Ans;
}

int main() {
    scanf("%d%d", &N, &M);
    int All = 0;
    for (int i = 1; i <= N; i ++) {
        scanf("%d", &Num[i]);
        All += Num[i];
    }
    for (int i = 1; i <= N; i ++) P[i] = 1.0 * Num[i] / All;
    double Ans = 1.0 * M * (M - 1);
    double Sum = 0;
    for (int i = 1; i <= N; i ++) Sum += P[i] * P[i];
    Ans = Ans * Sum + M;
    printf("%.2lf\n", Ans);
    Ans = N;
    for (int i = 1; i <= N; i ++) Ans -= Power(1 - P[i], M);
    printf("%.2lf\n", Ans);
}