n个骰子的点数之和的次数

题目：把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个骰子点数和及各自出现的概率