题目
题目描述: 把n个骰子扔在地上,所有骰子朝上一面的点数之和为S。输入n,打印出S的所有可能的值出现的概率。 输入: 输入包括一个整数N(1<=N<=1000),代表有N个骰子。 输出: 可能有多组测试数据,对于每组数据, 按照Sample Output的格式输出每一个可能出现的和S的概率。
前言
这道题目在《剑指offer》上有原题,但是感激它上面分析的有些浅,起码连这个题目属于哪种类型都没有说出来,下面我讲一下我对这道题目的理解
解题思路
首先,我判断这是一道动态规划的题目,因为
有重叠子问题,一个子问题在下一决策中会被多次用到
我们设有k个骰子,点数和为n时,n出现的次数为f(k,n),则可以分析出与k-1个骰子阶段的关系为:
当我有k-1个骰子时,再增加一个骰子,这个骰子的点数只可能为1,2,3,4,5,6.那有k个骰子得到点数和为n只有下述的6种情况:
- (k-1,n-1):第k个骰子投了点数1
- (k-1,n-2):第k个骰子投了点数2
- (k-1,n-3):第k个骰子投了点数3
- (k-1,n-4):第k个骰子投了点数4
- (k-1,n-5):第k个骰子投了点数5
- (k-1,n-6):第k个骰子投了点数6
所以,f(k,n) = f(k-1,n-1) + f(k-1,n-2) + f(k-1,n-3)+ f(k-1,n-4) + f(k-1,n-5) + f(k-1,n-6)
初始情况下,f(1,1) = f(1,2) = f(1,3) = f(1,4) = f(1,5) = f(1,6) =1
动态规划的标准代码
#include <stdio.h>
#include <stdlib.h>
#include <math.h>
int main(void)
{
int i, j, k, n, row, column;
double count, **arr;
while (scanf("%d", &n) != EOF) {
// 动态申请二维数组
row = n;
column = 6 * n;
arr = (double **)malloc((row + 1) * sizeof(double *)); // 定义第一列(即每一行首地址)
for (i = 0; i <= row; i ++) {
arr[i] = (double *)malloc((column + 1) * sizeof(double)); // 定义每一行
}
// 赋初值
for (i = 1; i <= column; i ++) {
if (i <= 6)
arr[1][i] = 1.000;
else
arr[1][i] = 0.000;
}
// 动态规划
for (i = 2; i <= row; i ++) {
for (j = 1; j <= column; j ++) {
if (j > i * 6 || j < i) { // 超过最大or小于最小
arr[i][j] = 0.000;
} else {
for (k = j - 6, arr[i][j] = 0; k < j; k ++) {
if (k < 1)
continue;
else
arr[i][j] += arr[i - 1][k];
}
}
}
}
// 打印输出
for (i = row, count = pow(6, row); i <= column; i ++) {
printf("%d %.3f
", i, arr[row][i] / count);
}
printf("
");
}
return 0;
}
但是,九度提交竟然提示我超内存限制:
这道题内存限制为32M,也就是32768kb,我超出了72kb,蛋疼!
AC代码
考虑行数为2,用%的方法来保存上次的结果集即可,ac代码如下:
#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#define MAX 6
int main()
{
int i, j, k, n;
double count, **arr;
while (scanf("%d", &n) != EOF) {
// 动态申请二维数组
arr = (double **)malloc(sizeof(double *) * 2);
for (i = 0; i < 2; i ++) {
arr[i] = (double *)malloc(sizeof(double) * (MAX * n + 1));
}
for (i = 1; i <= 6; i ++) // 初始化
arr[1][i] = 1.000;
for (i = 2; i <= n; i ++) {
for (j = i; j <= i * 6; j ++) {
arr[i % 2][j] = 0.000;
for (k = j - 6; k < j; k ++) {
if (k > 0) {
arr[i % 2][j] += arr[(i + 1) % 2][k];
}
}
}
}
for (count = pow(6, n), i = n; i <= n * 6; i ++)
printf("%d: %.3f
", i, arr[n % 2][i] / count);
printf("
");
free(arr);
}
return 0;
}
/**************************************************************
Problem: 1255
User: wangzhengyi
Language: C
Result: Accepted
Time:290 ms
Memory:3404 kb
****************************************************************/