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]]
。
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;
}
}
}