题目:把n个骰子扔在地上,所有骰子朝上一面的点数之和为s。输入n,打印出k的所有可能的值出现的概率。
思路:基于循环求骰子点数,时间性能好。
在一次循环中,第一个数组中的第n个数字表示骰子和为n出现的次数。在下一次循环中,我们加上一个新的骰子,此时和为n的骰子出现的次数等于上一次循环中骰子点数和为n-1、n-2、n-3、n-4、n-5和n-6的次数的总和,所以我们把另一个数组的第n个数字设为前一个数组对应的第n-1、n-2、n-3、n-4、n-5与n-6的和。
public int DiceProbability(int n, int sum) {
if (n == 0 || sum > n*6) // 无法得到sum的情况
return -1;
int[] s = new int[n*6+1];
int k = 1;
for (int i=1; i<=6; i++) // 初始 1 个骰子
s[i] = 1;
while (k < n) {
int[] tmp = new int[n*6+1]; // 用来暂存新一轮和
for(int i=k; i<=(k+1)*6; i++) {
for (int p=1; p<=6; p++) {
if (i-p > 0) {
tmp[i] += s[i-p];
}
}
}
s = Arrays.copyOf(tmp, s.length); // 不能单个更新,算后面的和会出现错误
k ++;
}
return s[sum];
}