骰子求和

LintCode 20 骰子求和

扔 n 个骰子，向上面的数字之和为 S。给定 Given n，请列出所有可能的 S 值及其相应的概率。

样例

给定 n = 1，返回 [ [1, 0.17], [2, 0.17], [3, 0.17], [4, 0.17], [5, 0.17], [6, 0.17]]。

 

起初打算用sum迭代按顺序增加每个骰子的结果，最后以sum为下标将结果数组加上一次基数，

代码：

public class Solution {
    /**
     * @param n an integer
     * @return a list of Map.Entry<sum, probability>
     */
    private static double basePro;
    public List<Map.Entry<Integer, Double>> dicesSum(int n) {
		basePro = (double)1/Math.pow(6,n);
        List<Map.Entry<Integer, Double>> resultsList = new ArrayList<Map.Entry<Integer, Double>>();
        double[] results = new double[6*n];
        doAdd(results,n,1,0);
        for(int i=n;i<=6*n;i++) {
        	resultsList.add(new AbstractMap.SimpleEntry<Integer, Double>(i, results[i-1]));
        }
        return resultsList;
    }

	private void doAdd(double[] results, int n, int index,int sum) {
		if(index==n+1) {
			results[sum-1]+=basePro;
			return;
		}
		for(int i=1;i<=6;i++) {
			doAdd(results, n, index+1, sum+i);
		}
	}
}

运行到15时超时，时间复杂度太大，ide也运行不了。

 

解：

  动态规划问题

思路：

  初始化一个骰子的情况。

  从第二个骰子开始，当计算i个骰子sum为j的结果时，第i个骰子有6种结果，想要获得结果j，需要从i-1个骰子结果集中找到相应的结果加上第i个骰子的结果，即前六个结果。对应表格：

x：总点数 y：骰子数 表格：情况数

					+6	+5	+4	+3	+2	+1
	1	2	3	4	5	6	7	8	9	10	11	12	13	14	15	16	17	18
1	1	1	1	1	1	1
2		1	2	3	4	5	6	5	4	3	2	1
3			1	3	6	10	15	21	25	27	27	25	21	15	10	6	3	1

 

代码：

package test;

import java.util.AbstractMap;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

public class Solution20 {

	public List<Map.Entry<Integer, Double>> dicesSum(int n) {
        List<Map.Entry<Integer, Double>> resultsList = new ArrayList<Map.Entry<Integer, Double>>();
        double[][] results = new double[n+1][6*n+1];
        results[1][1]=results[1][2]=results[1][3]=results[1][4]=results[1][5]=results[1][6]=(double)1/6;
        for(int i=2;i<=n;i++) {
        	for(int j=i;j<=6*i;j++) {
        		for(int k=0;k<6;k++) {
                    //寻找当前骰子每一个点数对应的结果
        			if(j-k>0)//前方可能无结果
        				results[i][j]+=results[i-1][j-k];
        		}
        		results[i][j]/=6;
        		//每一层除以6，及对应/6，/36。。概率
        	}
        }
        for(int i=n;i<=6*n;i++)
        	resultsList.add(new AbstractMap.SimpleEntry<Integer, Double>(i, results[n][i]));
        return resultsList;
    }

}