题目:把n个骰子扔在地上,所有骰子朝上一面的点数之和为S。输入n,打印出S的所有可能的值出现的概率。
解题思路:
第一步,确定问题解的表达式。可将f(n, s) 表示n个骰子点数的和为s的排列情况总数。
第二步,确定状态转移方程。n个骰子点数和为s的种类数只与n-1个骰子的和有关。因为一个骰子有六个点数,那么第n个骰子可能出现1到6的点数。所以第n个骰子点数为1的话,f(n,s)=f(n-1,s-1),当第n个骰子点数为2的话,f(n,s)=f(n-1,s-2),…,依次类推。在n-1个骰子的基础上,再增加一个骰子出现点数和为s的结果只有这6种情况!那么有:
f(n,s)=f(n-1,s-1)+f(n-1,s-2)+f(n-1,s-3)+f(n-1,s-4)+f(n-1,s-5)+f(n-1,s-6) ,0< n<=6n
f(n,s)=0, s< n or s>6n
上面就是状态转移方程,已知初始阶段的解为:
当n=1时, f(1,1)=f(1,2)=f(1,3)=f(1,4)=f(1,5)=f(1,6)=1。
递归版本:
#include <vector>
#include <iostream>
#include <math.h>
using namespace std;
int fun(int n, int count) {
if (count < n || n< 1 || count > 6*n)
return 0;
if (n == 1)
return 1;
int ans = fun(n - 1, count - 1) + fun(n - 1, count - 2) + fun(n - 1, count - 3)
+ fun(n - 1, count - 4) + fun(n - 1, count - 5) + fun(n - 1, count - 6);
return ans;
}
int main() {
int n;
cin >> n;
vector<int> data(5 * n + 1);
vector<int> res(5 * n + 1);
for (int i = 0; i <= 5 * n; i++) {
data[i] = i+n;
res[i] = fun(n, data[i]);
cout << data[i] << " " << res[i] << endl;
}
system("pause");
return 0;
}
迭代版本:
int getNSumCountNotRecusion(int n, int* count){
if(n<1)
return -1;
//初始化最初状态
count[0]=count[1]=count[2]=count[3]=count[4]=count[5]=1;
if(n==1) return 0;
for(int i=2;i<=n;++i){
for(int sum=6*i;sum>=i;--sum){
int tmp1=((sum-1-(i-1))>=0?count[sum-1-(i-1)]:0); //上一阶段点数和sum-1的排列总数
int tmp2=(sum-2-(i-1)>=0?count[sum-2-(i-1)]:0);
int tmp3=(sum-3-(i-1)>=0?count[sum-3-(i-1)]:0);
int tmp4(sum-4-(i-1)>=0?count[sum-4-(i-1)]:0);
int tmp5=(sum-5-(i-1)>=0?count[sum-5-(i-1)]:0);
int tmp6=(sum-6-(i-1)>=0?count[sum-6-(i-1)]:0);
count[sum-i]=tmp1+tmp2+tmp3+tmp4+tmp5+tmp6;
}
}
return 0;
}
//验证
int main(){
int n;
while(true){
cout<<"iteration input dice num:";
cin>>n;
int* count=new int[5*n+1];
memset(count,0,(5*n+1)*sizeof(int));
getNSumCountNotRecusion(n,count);
int allCount=0;
for(int i=0;i<5*n+1;++i){
cout<<"f("<<n<<","<<i+n<<")="<<count[i]<<endl;
allCount+=count[i];
}
delete[] count;
}
}
参考文献
【1】剑指Offer.何海涛.电子工业出版社.
【2】n个骰子点数和及各自出现的概率