1. 程式人生 > >Lintcode 20 骰子求和

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]]

思路:扔n個骰子數字和及概率相當於求前n-1個骰子和及概率與第n個骰子和及概率。
1.當n<=0,返回空
2.當n=1時,返回1-6概率0.17;
3.當n>1時,遞迴呼叫返回List<map.entry>,對List的每一個骰子和
(Entry中的key)迴圈加1-6,作為新的key存到一個map中,
①當map中不包含這個key時,Map中插入概率等於1.0/6*value(value是Entry中的value)

②當map中包含這個key時,Map中插入概率等於之前的概率+1.0/6*value

程式碼如下:

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
        HashMap<Integer, Double> map = new HashMap<Integer, Double>();

        List<Map.Entry<Integer, Double>> list = null;


        
        if(n<=0){
           return null;
        }
        if(n ==1){
            for(int i =1;i<=6;i++){
            map.put(i,1.0/6);
            }
            
            list = new ArrayList<Map.Entry<Integer,Double>>(map.entrySet());
            
            return list;
        }else{
            list = dicesSum(n-1);
            System.out.println(list);
            int size = list.size();
            for(int i=0;i<size;i++){
                
                Map.Entry<Integer,Double> mapentry = list.get(i);
                
                Integer key = mapentry.getKey();
                Double value = mapentry.getValue();
                Integer tempkey =0;
                Double tempvalue = 0.0;
                for(int j=1;j<=6;j++){
                    tempkey = key+j;;
                    if(!map.containsKey(tempkey)){
                        tempvalue= value*(1.0/6);
                    }else{
                        tempvalue = value*1.0/6+map.get(tempkey);
                    }
                
                    map.put(tempkey,tempvalue);
                }
                
                
            }
            list = new ArrayList<Map.Entry<Integer,Double>>(map.entrySet());
            return list;
        }
    }
}