面试算法(四十三)n个骰子的点数

来源:互联网 发布:c语言输入保留两位小数 编辑:程序博客网 时间:2024/06/02 21:44

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

n个骰子的点数和的最小值为n,最大值为6n。n个骰子的所有点数的排列数为6的n次方。


解法一:基于递归求骰子点数,时间效率不高。

先考虑如何统计每一个点数出现的次数。先把n个骰子分成两堆:第一堆只有一个,第二堆有n-1个。单独的那一个有可能出现从1到6的点数。计算从1到6的每一种点数和剩下的n-1个骰子来计算点数和。

再把第二堆分成两堆,第一堆只有一个,第二堆有n-2个。把上一轮那个单独骰子的点数和这一轮单独骰子的点数相加,再和剩下的n-2个来计算点数和。

这是递归形式,结束条件是最后只剩下一个骰子。我们可以定义一个长度为6n-n+1的数组,和为s的点数出现的次数保存到数组第s-n个元素里。

int g_maxValue = 6;void PrintProbability(int number){if(number < 1)return;int maxSum = number * g_maxValue;int* pProbabilities = new int[maxSum - number + 1];for(int i=number; i<=maxSum; ++i)pProbabilities[i-number] = 0;Probability(number, pProbabilities);int total = pow((double)g_maxValue, number);for(int i=number; i<=maxSum; ++i){double ratio = (double)pProbabilities[i-number] / total;printf("%d: %e\n", i, ratio);}delete[] pProbabilities;}void Probability(int number, int* pProbabilities){for(int i=1; i<=g_maxValue; ++i)Probability(number, number, i, pProbabilities);}void Probability(int original, int current, int sum, int* pProbabilities){if(current == 1){pProbabilities[sum - original]++;}else{for(int i=1; i<=g_maxValue; ++i){Probability(original, current-1, i+sum, pProbabilities);}}}

解法二:基于循环求骰子点数,时间性能好。

用两个数组来存储骰子点数的每一个总数出现的次数。在一次循环中,第一个数组中的第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之和。

int g_maxValue = 6;void PrintProbability(int number){if(number < 1)return;int pProbabilities[2];pProbabilities[0] = new int[g_maxValue * number + 1];pProbabilities[1] = new int[g_maxValue * number + 1];for(int i=0; i<g_maxValue * number + 1; ++i){pProbabilities[0][i] = 0;pProbabilities[1][i] = 0;}int flag = 0;for(int i=1; i<=g_maxValue; ++i)pProbabilities[flag][i] = 1;for(int k=2; k<=number; ++k){for(int i=0; i<k; ++i)pProbabilities[1-flag][i] = 0;for(int i=k; i<=g_maxValue * k; ++i){pProbabilities[1-flag][i] = 0;for(int j=1; j<=i  && j<=g_maxValue; ++j)pProbabilities[1-flag][i] += pProbabilities[flag][i-j];}flag = 1- flag;}double total = pow((double)g_maxValue, number);for(int i=nuber; i<=g_maxValue * number; ++i){double ratio = (double)pProbabilities[flag][i] / total;printf("%d: %e\n", i, ratio);}delete[] pProbabilities[0];delete[] pProbabilities[1];}


0 0