cubegao

Swift.n个骰子的点数

2015-06-24

题目描述:把 n 个骰子扔在地上,所有骰子朝上一面的点数之和为 s。输入 n,打印出 s 的所有可能的值出现的概率。

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
import Foundation

class For43Solution {
func diceProbility(diceCount:Int) {

var arr = [[Int]]()
let maxCount = 6 * diceCount + 1
let first = [Int](repeating: 0, count: maxCount)
let next = [Int](repeating: 0, count: maxCount)
arr.append(first)
arr.append(next)

var flag = 0

//一个骰子的情况
for i in 1...6 {
arr[flag][i] = 1
}

//剩下的骰子
for i in 2...diceCount {
//清空不存在的点数,2个骰子不存在1
for j in 0..<i {
arr[1-flag][j] = 0
}
//计算每个存在的点数
for k in i...6*i {
var m = 1
arr[1-flag][k] = 0
while m <= 6 && m <= k {
arr[1-flag][k] += arr[flag][k-m]
m += 1
}
}

flag = 1 - flag
}

//打印
let total :Double = pow(6.0, Double(diceCount))
var index = 0
for i in arr[flag] {
print("点数\(index)出现的概率:\(Double(i)/total)")
index += 1
}

}

}

算法思想:动态规划求解,每次都算出可能存在的所有点数。

github地址:https://github.com/cubegao/LeetCode

Tags: 算法

扫描二维码,分享此文章