面试题60. n个骰子的点数

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

输入: 1
输出: [0.16667,0.16667,0.16667,0.16667,0.16667,0.16667]

示例 2:

输入: 2
输出：[0.02778,0.05556,0.08333,0.11111,0.13889,0.16667,0.13889,0.11111,0.08333,0.05556,0.02778]

限制：1 <= n <= 11

这个题一定要把方向把握准确，我最开始考虑问题的角度是如何能扔出和为n的点数，然后就进入了死胡同，最后看了解析是动态规划，也是想不明白为啥，最后看了2天，终于想通了为啥，那我们来看看吧。

(一）动态规划（二维dp)

这个问题我们不要去考虑如何得到点数，这就是数学问题了，你要解决就要进行扔骰子模拟编码，这个就是死胡同了，我们需要关注的是点数n出现了几次，最后除以总共的次数就可以算出概率了。

我们先来分析分析，当只有一个骰子时，点数1，2，3，4，5，6的概率均为1/6，它的点数范围是1~6，总共出现点数的可能性是6^1。那么我们有n个骰子的话，点数范围就是n~6n,总共出现点数的情况是6^n。
我们现在把当骰子数为1个和2个的情况列出来，便于我们分析：
我们可知：骰子数为2个时：数值范围为2~12，总共有36种情况，所以我们求出每一个数值出现的次数，除以36即为概率：

那我们就根据分析，理一下代码思路：

• 先申请二维数组dp，初始化0，再给一个骰子dp[1][1]~dp[1][6]赋值为1；
• 第一个for循环从i=2开始，给dp[2][j]到dp[n][j]赋值；第二个循环是给n个骰子的数值范围赋值，从n~6n；第三个循环是累加给dp[n][j]赋值。
• 最后我们将第n行的次数/6^n，得到概率，存入数组res。

vector<double> twoSum(int n) 
{
	vector<double> res;//存储概率结果的
	vector<vector<double>> dp(n+1,vector<double>(6*n+1,0));//dp数组初始化为0

	//初始化1枚骰子
	for(int i=1;i<=6;i++)
	{
		dp[1][i]=1;
	}
	//第一个for,n个骰子，每一行的填充
	for(int i=2;i<=n;i++)
	{
		//第2个for,进行可能出现的点数遍历，列的填充
		for(int j=i;j<=6*i;j++)
		{
			//第3个for,累加上一次结果，给dp[i][j]赋值
			for(int k=1;k<=6;k++)
			{
				if(j-k<=0)//超出范围
					break;
				dp[i][j]+=dp[i-1][j-k];
			}
		}
	}
	//把次数/总次数，算出概率，压入res
	for(int i=n;i<=6*n;i++)
	{
		double tmp=dp[n][i]*1.0/(pow(6,n));
		res.push_back(tmp);
	}
	return res;
}
int main()
{
	vector<double> res;
	res=twoSum(2);
	for(int i=0;i<res.size();i++)
	{
		cout<<res[i]<<" ";
	}
}

这个时间复杂度为O(N^2),空间为O(N^2）。我们发现每一次的结果只和上一阶段有关，那么我们可以改进为一维数组。

(二）改进为一维dp

我们进行空间复杂度的优化，主要思想如下：

那么我们代码的思路：

• 定义dp[6n+1]，初始化数组dp；
• 第一个for循环控制骰子次数，因为n个骰子的结果是由n-1个骰子得到的，所以它从i=2开始计算；第二个循环给骰子可能的结果赋值，从后往前，即6n~n；第三个循环是累加给dp[n][j]赋值。
• 一定要记得dp[j]每次计算前先清0，否则就会出现dp[6]本身就为1，不清0，就会在1的基础上开始加，那么结果错误。

vector<double> twoSum(int n) 
{
	vector<double> res;//存储概率结果的
	vector<double> dp(6*n+1,0);//dp数组初始化为0
	//初始化1枚骰子
	for(int i=1;i<=6;i++)
	{
		dp[i]=1;
	}
	//第一个for,n个骰子，
	for(int i=2;i<=n;i++)
	{
		//第2个for,从后往前进行可能出现的点数遍历，
		for(int j=6*i;j>=i;j--)
		{
			dp[j]=0;//本身要先清0
			//第3个for,进行累加
			for(int k=1;k<=6;k++)
			{
				if(j-k<i-1)//数值范围：i~6i
					break;
				dp[j]+=dp[j-k];
			}
		}
	}
	//把次数/总次数，算出概率，压入res
	for(int i=n;i<=6*n;i++)
	{
		double tmp=dp[i]*1.0/(pow(6,n));
		res.push_back(tmp);
	}
	return res;
}

时间复杂度为O(N^2),空间为O(N)。

加油哦！💪。