n个骰子的点数【每日一练】

1. 思想

问题转化
这个问题可以先转化为求总点数之和对应的次数，然后用次数除以总次数（6^n) 就可以算出概率；

动态规划求解
计算每个总和出现的次数可以用动态规划，数组 dp[i] 表示在当前骰子数下，总和为 i 的出现次数。每增加一个骰子，dp[i] 可以从上一个 dp 转移过来

2. 算法代码

vector<double> dicesProbability(int n) {
		// 骰子有多少面
		int Face = 6;
	  int faceSum = Face*n;
		// 1. 利用 dp 求出每个总数和对应的出现次数
		
		// newdp 代表当前的骰子总和，olddp 相当于一个临时数组，用于在计算下一次 dp 时保存旧 newdp
	  vector<int> newdp(faceSum+1, 0), olddp(faceSum+1, 0);
	  fill(newdp.begin()+1, newdp.begin()+Face+1, 1);
	  for(int i=2; i<=n; ++i){
	      olddp.swap(newdp);
			// 注意，这里必须是从 1 开始计算 dp 数组（2也可以，但不能是 n）
			// 因为骰子在只有 2 颗的时候，总和最小是 2，所以必须考虑在内
	      for(int j=1; j<=Face*n; ++j){
	          for(int face=1; face<=6; ++face){
	              newdp[j] += j-face > 0 ? olddp[j-face] : 0;
	          }
	      }
	      fill(olddp.begin(), olddp.end(), 0);
	  }
		
		// 2. 计算概率，范围在 [n, Face*n]
	  vector<double> ans;
	  double cnt = pow(Face, n);
	  for(int i=n; i<=Face*n; ++i){
	      ans.push_back(newdp[i]*1.0/cnt);
	  }
	  return ans;
}

• 注意：在计算 dp 的时候，要从 1 开始，$Face\ast n$（或者 $Face\ast i$，$i$ 是当前的骰子数) 结束

3. 附录

#include <vector>
using namespace std;

题目：n 个骰子的点数

• 描述：把n个骰子扔在地上，所有骰子朝上一面的点数之和为s。输入n，打印出s的所有可能的值出现的概率。
  你需要用一个浮点数数组返回答案，其中第 i 个元素代表这 n 个骰子所能掷出的点数集合中第 i 小的那个的概率。