十一城

跬步千里,小流江海。

Home Linux ML Python Java Thoughts KmKg BookCan Links About

2018-05-03
算法-dp-骰子求和


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
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;

}
}

dzzxjl

Home Linux ML Python Java Thoughts KmKg BookCan Links About