一、题目

把n个骰子扔在地上,所有骰子朝上一面的点数之和为s。输入n,打印出s 的所有可能的值出现的概率。

二、解题思路

解法一:基于通归求解,时间效率不够高。

先把n个骰子分为两堆:第一堆只有一个,另一个有n- 1 个。单独的那一个有可能出现从1 到6 的点数。我们需要计算从1 到6 的每一种点数和剩下的n-1 个骰子来计算点数和。接下来把剩下的n-1个骰子还是分成两堆,第一堆只有一个, 第二堆有n-2 个。我们把上一轮那个单独骰子的点数和这一轮单独骰子的点数相加, 再和剩下的n-2 个骰子来计算点数和。分析到这里,我们不难发现这是一种递归的思路,递归结束的条件就是最后只剩下一个骰子。

我们可以定义一个长度为6n-n+1 的数组, 和为s 的点数出现的次数保存到数组第s-n个元素里。

解法二:基于循环求解,时间性能好

我们可以考虑用二维数组来存储骰子点数的每一个总数出现的次数。在一次循环中, 第一个数组中的第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之和。

三、解题代码

  1. public class Test {
  2. /**
  3. * 基于通归求解
  4. *
  5. * @param number 色子个数
  6. * @param max 色子的最大值
  7. */
  8. public static void printProbability(int number, int max) {
  9. if (number < 1 || max < 1) {
  10. return;
  11. }
  12. int maxSum = number * max;
  13. int[] probabilities = new int[maxSum - number + 1];
  14. probability(number, probabilities, max);
  15. double total = 1;
  16. for (int i = 0; i < number; i++) {
  17. total *= max;
  18. }
  19. for (int i = number; i <= maxSum; i++) {
  20. double ratio = probabilities[i - number] / total;
  21. System.out.printf("%-8.4f", ratio);
  22. }
  23. System.out.println();
  24. }
  25. /**
  26. * @param number 色子个数
  27. * @param probabilities 不同色子数出现次数的计数数组
  28. * @param max 色子的最大值
  29. */
  30. private static void probability(int number, int[] probabilities, int max) {
  31. for (int i = 1; i <= max; i++) {
  32. probability(number, number, i, probabilities, max);
  33. }
  34. }
  35. /**
  36. * @param original 总的色子数
  37. * @param current 剩余要处理的色子数
  38. * @param sum 已经前面的色子数和
  39. * @param probabilities 不同色子数出现次数的计数数组
  40. * @param max 色子的最大值
  41. */
  42. private static void probability(int original, int current, int sum, int[] probabilities, int max) {
  43. if (current == 1) {
  44. probabilities[sum - original]++;
  45. } else {
  46. for (int i = 1; i <= max; i++) {
  47. probability(original, current - 1, i + sum, probabilities, max);
  48. }
  49. }
  50. }
  51. /**
  52. * 基于循环求解
  53. * @param number 色子个数
  54. * @param max 色子的最大值
  55. */
  56. public static void printProbability2(int number, int max) {
  57. if (number < 1 || max < 1) {
  58. return;
  59. }
  60. int[][] probabilities = new int[2][max * number + 1];
  61. // 数据初始化
  62. for (int i = 0; i < max * number + 1; i++) {
  63. probabilities[0][i] = 0;
  64. probabilities[1][i] = 0;
  65. }
  66. // 标记当前要使用的是第0个数组还是第1个数组
  67. int flag = 0;
  68. // 抛出一个骰子时出现的各种情况
  69. for (int i = 1; i <= max; i++) {
  70. probabilities[flag][i] = 1;
  71. }
  72. // 抛出其它骰子
  73. for (int k = 2; k <= number; k++) {
  74. // 如果抛出了k个骰子,那么和为[0, k-1]的出现次数为0
  75. for (int i = 0; i < k; i++) {
  76. probabilities[1 - flag][i] = 0;
  77. }
  78. // 抛出k个骰子,所有和的可能
  79. for (int i = k; i <= max * k; i++) {
  80. probabilities[1 - flag][i] = 0;
  81. // 每个骰子的出现的所有可能的点数
  82. for (int j = 1; j <= i && j <= max; j++) {
  83. // 统计出和为i的点数出现的次数
  84. probabilities[1 - flag][i] += probabilities[flag][i - j];
  85. }
  86. }
  87. flag = 1 - flag;
  88. }
  89. double total = 1;
  90. for (int i = 0; i < number; i++) {
  91. total *= max;
  92. }
  93. int maxSum = number * max;
  94. for (int i = number; i <= maxSum; i++) {
  95. double ratio = probabilities[flag][i] / total;
  96. System.out.printf("%-8.4f", ratio);
  97. }
  98. System.out.println();
  99. }
  100. }