n个色子的点数

问题：

n个色子，每个色子m面，每一面的值分别为m-1。你将n个色子同时抛，落地后将所有朝上面的数字加起来，记为sum。给定一个数字x，如果sum>x，则你赢。给定n，m，x，求你赢的概率。

分析：

何海涛的《剑指offer》中有类似的题目，但这里问的更Gerneral。

要求的是概率，所以可以从两个方面入手：种类数 & 概率。

PS：认为每个面的值为 0,1,...,m-1。

一、从种类数入手：

总的种类数为m^n，只需求出其中满足sum>x的个数K，即可得到概率: K / (m^n)。

方法I：暴力枚举

枚举每个色子的点数，判断是否大于x。

此方法存在重复计算，复杂度为指数级。

方法II：动态规划

用f[i,k]表示前i个色子掷出的点数和为k的种数。那么：

f[i+1,k] = Σ f[i, k-j] ， j = 0,1,...,m-1。

算法可以使用滚动数组记录：开两个长度为 n*（m-1）+1的数组即可。

还可以仿照01背包的实现：逆序求和。那样的话，就只需要开一个数组了。

代码：时间复杂度O(n * nm)，空间复杂度O（nm)。

double getProperity(int n, int m, int x){
	int sum = (m - 1) * n;
	if(sum <= x) return 0;
	vector<int> count(sum+1, 0);
	fill_n(count.begin(), m, 1);//只有一个色子时
	int maxSum = m - 1;
	for(int dicei = 1; dicei < n; ++dicei){
		maxSum += m-1;
		int counti = accumulate(count.begin() + maxSum + 1 - m, 
			count.begin() + maxSum + 1, 0);//统计前m个的和
		for(int sumi = maxSum; sumi > 0; --sumi){//逆序遍历，只用一个数组即可
			int tmp = count[sumi];
			count[sumi] = counti;
			counti -= tmp;
			counti += sumi - m < 0 ? 0 : count[sumi - m];
		}
	}
	return accumulate(count.begin() + x + 1, count.end(), 0.0) / pow((double)m, (double)n);
}

此代码不使用滚动数组、不用遍历m个点数，在时间和空间上都比《剑指offer》给出的代码实现要 更优。

二、直接从概率入手：

仍然是动态规划思想：

用f[i,k]表示前i个色子掷出的点数和为k的概率。那么：

f[i+1,k] = Σ f[i, k-j] / m， j = 0,1,...,m-1。

代码：与上面的实现类似，时间复杂度O(n * nm)，空间复杂度O（nm)。

double getProperity2(int n, int m, int x){
	int sum = (m - 1) * n;
	if(sum <= x) return 0;
	vector<double> properity(sum+1, 0.0);
	fill_n(properity.begin(), m, 1.0/m);//只有一个色子时
	int maxSum = m - 1;
	for(int dicei = 1; dicei < n; ++dicei){
		maxSum += m-1;
		double properityi = accumulate(properity.begin() + maxSum + 1 - m, 
			properity.begin() + maxSum + 1, 0.0);//统计前m个的和
		for(int sumi = maxSum; sumi >= 0; --sumi){//逆序遍历，只用一个数组即可
			double tmp = properity[sumi];
			properity[sumi] = properityi / m;
			properityi -= tmp;
			properityi += sumi - m < 0 ? 0 : properity[sumi - m];
		}
	}
	return accumulate(properity.begin() + x + 1, properity.end(), 0.0);
}