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

题目大意

$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[\Delta_i]=E[2m_i+1]=2E[m_i]+1\\ E[m_i]=t*p_i$$
所以 $E[\Delta_i]=2t*p_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$的情况加起来就可以了。
$$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$$
同样对于第二问也十分好求，由于期望的线性性，我们考虑对每个人分开考虑然后加起来，显然
$$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);
}