n个骰子的点数20210308

题目

把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个骰子时，点数和j出现的次数；我们可以用数组的第一维表示阶段，即已经扔了几个骰子，数组的第二维表示扔完这些骰子后可能出现的点数和；而数组值就是该阶段下点数和出现的次数；

第二步：状态转移

既然确定了状态，即二维数组dp[n][j]；那么接下来就是确认状态转移，即数组的演变关系；单单看第n枚骰子，它可能出现的点数是1,2，3,4，5,6，因此扔完n枚骰子后点数和j出现的次数可以由扔完第n-1枚骰子后，对应点数和j-1,j-2,j-3,j-4,j-5,j-6出现的次数之和转化过来；【注意理解这块逻辑】

第三步：找边界条件

• 第一阶段的点数j分别是1,2，3,4，5,6，且出现的次数都是1；
• n枚筛子，点数和j的取值范围[n,6*n]；

代码

时间复杂度：O(N^2)
空间复杂度：-

public double[] dicesProbability(int n) {
    // dp数组，为了更好的表达，n=0无实际意义
	int[][] dp = new int[n+1][6*n+1];
	// 初始化边界条件，即第一阶段的点数和为1到6，出现的次数都为1
	for (int i = 1; i <= 6; i++) {
		dp[1][i] = 1;
	}
	
	// 从第二阶段开始一直到第n阶段
	for (int i = 2; i <= n; i++) {
		// i枚骰子点数和j的取值范围是[i,6*i]
		for (int j = i; j <= 6*i; j++) {
			// 状态转移的代码表达，即第n阶段的点数和j出现的次数由第n-1节点点数和j-1,j-2...,j6的次数之和
			for (int cur = 1; cur <= 6; cur++) {
				if (j -cur <= 0) break;
				dp[i][j] +=dp[i-1][j-cur];
			}
		}
	}
	// n枚骰子总共会出现6的n次方个结果
	double num = Math.pow(6, n);
	// 最终的结果数组，长度为6*n - n +1
	double[] res = new double[5*n+1];
	// 根据次数除于总次数，得到结果
	for (int i = n; i <= 6*n; i++) {
		res[i-n] = dp[n][i]/num;
	}
	return res;
}