投n个骰子,第 i 个骰子有可能投掷出 Xi 种等概率的不同的结果,数字从 1 到 Xi。所有骰子的结果的至大至将作为最终结果。求最终结果的期望。
输入描述:
第一行一个整数n,表示有n个骰子。(1 <= n <= 50)
第二行 n 个整数,表示每个骰子的结果数 Xi。(2 <= Xi <= 50)
输出描述:
输出最终结果的期望。保留两位小数。
example:
输入
2
2 2
输出
1.75
先找出所有可能出现的结果值,对于每种结果值,计算该结果对于的概率。
假设所有结果的最大值为 i,对应的该概率为 p,首先找到所有投掷结果可能大于等于 i 的骰子,依次挑选每一个骰子 j 的结果一定为 i,其他骰子结果小于等于 i,为避免重复结果,让最大值小于骰子 j 最大值的筛子取值为小于 i,最大值大于骰子 j 最大值的筛子取值为小于等于 i。举例说明如下:
假设此时计算结果为 4 的概率。选中投掷结果一定为 4 的是骰子5。
骰子编号 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 |
---|---|---|---|---|---|---|---|---|
骰子最大值 | 2 | 3 | 4 | 5 | 5 | 6 | 7 | 10 |
可能取值 | 1,2 | 1,2,3 | 1,2,3 | 1,2,3 | 4 | 1,2,3,4 | 1,2,3,4 | 1,2,3,4 |
概率 | 1 | 1 | 3/4 | 3/5 | 1/5 | 4/6 | 4/7 | 4/10 |
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int main(){
int n;
cin >> n;
vector<int>vec(n, 0);
for (int i = 0; i < n; i++)
cin >> vec[i];
sort(vec.begin(), vec.end());
double res = 0;
int maxnum = vec[n - 1];
for (int i = 1; i <= maxnum; i++){ //每一种可能出现的最大值
double p = 0; //所有骰子最大值为i的概率
int pos = 0; //抛掷结果最大值大于等于i的开始位置
while (vec[pos] < i) pos++;
for (int j = pos; j < n; j++){ //挑选的位置,该位置结果为i,其他位置小于等于i
double pj = 1; //挑选位置j出现指定值的概率
for (int k = pos; k < n; k++){
if (j == k)
pj = pj * 1.0 / vec[k]; //挑选位置之前的抛掷结果等于i
else if (k > j)
pj = pj * (double)i / vec[k]; //挑选位置之后的抛掷结果大于等于i
else if (k < j)
pj = pj * (double)(i - 1) / vec[k]; //挑选位置之前的抛掷结果小于i,避免重复
}
p += pj; //总概率
}
res += i * p; //计算期望
}
printf("%.2f", res);
return 0;
}