Java實現-骰子求和

扔 n 個骰子,向上面的數字之和爲 S。給定 Given n,請列出所有可能的 S 值及其相應的概率。

 注意事項

You do not care about the accuracy of the result, we will help you to output results.

樣例

給定 n = 1,返回 [ [1, 0.17], [2, 0.17], [3, 0.17], [4, 0.17], [5, 0.17], [6, 0.17]]

public class Solution {
    /**
     * @param n an integer
     * @return a list of Map.Entry<sum, probability>
     */
    public List<Map.Entry<Integer, Double>> dicesSum(int n) {
        // Write your code here
        // Ps. new AbstractMap.SimpleEntry<Integer, Double>(sum, pro)
        // to create the pair
        long [][]dp=new long[n+1][6*n+1];
		dp[1][1]=1;
		dp[1][2]=1;
		dp[1][3]=1;
		dp[1][4]=1;
		dp[1][5]=1;
		dp[1][6]=1;
		for(int i=2;i<=n;i++){
			for(int j=i;j<=i*6;j++){
				long x1=0,x2=0,x3=0,x4=0,x5=0,x6=0;
				if(j-1>0){
					x1=dp[i-1][j-1];
				}
				if(j-2>0){
					x2=dp[i-1][j-2];
				}
				if(j-3>0){
					x3=dp[i-1][j-3];
				}
				if(j-4>0){
					x4=dp[i-1][j-4];
				}
				if(j-5>0){
					x5=dp[i-1][j-5];
				}
				if(j-6>0){
					x6=dp[i-1][j-6];
				}
				dp[i][j]=x1+x2+x3+x4+x5+x6;
			}
		}
		List<Map.Entry<Integer, Double>> result=new ArrayList<Map.Entry<Integer,Double>>();
		for(int i=n;i<=6*n;i++){
			AbstractMap.SimpleEntry<Integer, Double> entry=new AbstractMap.SimpleEntry<Integer, Double>(i, dp[n][i]/Math.pow(6, n));
			result.add(entry);
		}
		return result;
    }
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章