面试题60：n个骰子的点数

//面试题60：n个骰子的点数
//题目：把n个骰子扔在地上，所有骰子朝上一面的点数之和为s。输入n，打印出s
//的所有可能的值出现的概率。

public class DicesProbability {
    //方法一 递归
    int maxValue=6;
    public void PrintProbability(int number) {
        int maxSum=maxValue*number;
        int[] pProbability=new int[maxSum-number+1];

        Probability(number,pProbability);
        int total=(int) Math.pow(maxValue, number);
        for(int i=0;i<pProbability.length;i++) {
            double ratio=(double)pProbability[i]/total;
            System.out.println("总和"+(i+number)+"概率"+ratio);
        }

    }
    //计算n个骰子不同点数和的概率
    public void Probability(int number,int[] pProbability) {
        for(int i=1;i<=maxValue;i++) {
            //i看做第一次分出去的单独的骰子的点数，下面调用函数计算另一堆n-1的总和
            Probability(number,number,i,pProbability);
        }
    }
    //计算n-1个骰子不同点数和的概率
    public void Probability(int original,int current,int sum,int[] pProbability) {
        if(current==1) {//递归结束条件，未分出的current分到最后只剩一个，sum也加完了
            //填入出现次数，和为sum的出现次数保存在sun-original中
            pProbability[sum-original]++;
        }else {
            //再分出去一个骰子。点数为i
            for(int i=1;i<=maxValue;i++) {
            Probability(original,current-1,i+sum,pProbability);
            }
        }
    }

    //方法二，循环
    //换思路，用两个数组存值，在一轮循环中，第一个数组的第n个数存骰子和为n出现的次数，与上面不同
    //下一轮循环中，加上一个新的骰子，此时和为n的骰子出现的次数，应等于上一轮中和为n-1到n-6出现次数之和
    public void PrintProbability2(int number) {
        if(number<1) {
            return;
        }
        //定义两个长度为6n的数组，有浪费空间的嫌疑
        int[][] pProbability=new int[2][maxValue*number+1];
        //标记 两数据交替使用
        int flag=0;
        //先只有一个骰子,定义第一个数组值
        for(int i=1;i<=maxValue;i++) {
            pProbability[flag][i]=1;//1~6各只可能出现一次
        }
        //陆续加骰子,下一轮定义第二个数组值
        for(int k=2;k<=number;k++) {
            //骰子数为k时的总和
            for(int i=0;i<k;i++) {
                pProbability[1-flag][i]=0;//总和为0~k-1的次数为0
            }
            for(int i=k;i<=maxValue*k;i++) {
                pProbability[1-flag][i]=0;//清零，方便下面叠加运算
                for(int j=1;j<=i&&j<=maxValue;j++) {
                pProbability[1-flag][i]+=pProbability[flag][i-j];
                }
            }
            flag=1-flag;


        }
        int total=(int) Math.pow(maxValue, number);
        for(int i=number;i<=maxValue*number;i++) {
            double ratio=(double)pProbability[flag][i]/total;
            System.out.println("总和"+i+"概率"+ratio);
        }

    }
    public static void main(String[] agrs) {
        DicesProbability d=new DicesProbability();
        System.out.println("递归法");//一次减一个骰子
        d.PrintProbability(3);
        System.out.println("循环法");//一次加一个骰子
        d.PrintProbability2(3);
    }
}