剑指offer面试题43 n个骰子的点数

/*
题目是仍n个骰子 求各点数和出现的概率
概率计算么，一共是6^n 。。。然后计算各点数和出现的次数一除就行了么

但是这里 问题就在于怎么计算各点数出现的次数和。。一种是递归 一种是循环
递归 肯定效率不好。。
循环的话 想法很好。。
*/
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
using namespace std;
const int maxn=6;
//递归的方法 n到十以上 性能就差的不行了。。
//还有就死和我递归函数写的确实不好。。 这么简单的递归也想不出来。
void Probability(int number,int cur,int sum,int * prob)
{
    if(cur==1)
        prob[sum-number]++;
    else
    {
        for(int i=1;i<=maxn;++i)
            Probability(number,cur-1,sum+i,prob);
    }
}

void Probability(int number,int * prob)
{
    for(int i=1;i<=maxn;++i)
        Probability(number,number,i,prob);
}

void PrintProbability(int number)
{
    if(number<1) return ;
    int maxsum=maxn*number;
    int * prob=new int[maxsum-number+1];
    memset(prob,0,sizeof(prob));
    Probability(number,prob);

    int tot=pow((double)maxn,number);
    for(int i=number;i<=maxsum;++i)
    {
        double ratio=(double)prob[i-number]/tot;
        printf("%d %lf\n",i,ratio);
    }
    delete [] prob;

}
/*
下面开始写循环的，其实也有一点DP的思想
假如我们现在知道扔n-1个骰子 各点数出现的次数的数组了。。prob[i]就是i出现的次数
那么我们再扔一个骰子 也就其实是扔n个。 i点数出现的次数就是从i-1到i-6之间所有次数的和。
这样来回只需要两个数组 每次从一个数组推倒另一个数组 然后标记每次是该往哪个数组里存就可以了
这样性能就好的多了
*/
void PrintProbability_2(int num)
{
    if(num<1) return ;
    int * prob[2];
    prob[0]=new int[maxn*num+1];
    prob[1]=new int[maxn*num+1];
    for(int i=1;i<=maxn*num;++i)
        prob[0][i]=prob[1][i]=0;//这里有一个神奇bug 一开始忘记加i了 就是prob[0]=0 0就是NULL这样就把指针清空了 所以后面错了
    int flag=0;
    for(int i=1;i<=maxn;++i)
        prob[0][i]=1;
    for(int i=2;i<=num;++i)
    {
        for(int j=0;j<i;++j)//因为扔i个骰子 最小合是i 所以i前面是0
            prob[1-flag][j]=0;
        for(int j=i;j<=maxn*i;++j)
        {
            prob[1-flag][j]=0;
            for(int k=1;k<j && k<=maxn ;++k)
            prob[1-flag][j]+=prob[flag][j-k];
        }
        flag=1-flag;
    }

    double tot=pow(double(maxn),num);
    for(int i=num;i<=maxn*num;++i)
    {
        double ratio=(double)prob[flag][i]/tot;
        printf("%d %lf\n",i,ratio);
    }
    delete[] prob[0];
    delete[] prob[1];
}

int main()
{
    int n;
    while(cin>>n)
    {
        //PrintProbability(n);
        cout<<endl;
        PrintProbability_2(n);
    }
    return 0;
}