Kids and Prizes (概率dp求期望)

Kids and Prizes

SGU - 495

大致题意:

有n个奖品，m个人排队来选礼物，对于每个人，他打开的盒子，可能有礼物，也有可能已经被之前的人取走了，然后把盒子放回原处

求最后m个人取走礼物的期望总数

---

解题思路:

因为每个人打开礼物之后会放回原处,所以m个人选礼物是独立的

对于每个礼物不被人选中的概率是 ((n-1)/n)^m

那么对于所有礼物来说,不被人选中的礼物数的期望是 n*((n-1)/n)^m

根据容斥原理 ,m个人取走礼物的期望就是 n-n*((n-1)/n)^m

---

AC代码:

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
#define debug(a) cout << #a << " = " << a << endl;
using namespace std;
typedef long long ll;
const int N = 1e6 + 10;
int n, m;
int main(void)
{
	while (~scanf("%d%d", &n, &m)) {
		double p = (double)(n - 1) / n;
		double res = n - n * pow(p, m);
		printf("%.9f\n", res);
	}
	return 0;
}

---