Swift学习——n个骰子的总和

前端之家收集整理的这篇文章主要介绍了Swift学习——n个骰子的总和前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

参考自:http://www.cnblogs.com/python27/archive/2011/11/26/2263332.html
(作者原话)题目:把n个骰子仍在地上,所有骰子的点数和为s。输入n,打印s所有可能取值的概率。

分析1:容易知道,有n个骰子的话,s的最小取值为n(全为1),最大取值为6n(全为6)。

如果只有1个骰子,那么很简单,s取1,2,3,4,5,6的情况数均为1,概率为1/6;设想有n个骰子,出现和为s,我们可以这样考虑,如果第一个骰子有6中情况,取1,2,3,4,5,6;那么剩下的n-1个骰子的和则分别取s-1,s-2,s-3,s-4,s-5,s-6,我们将所有这些情况相加,就可以得出总的情况数。看出了吗?亲,这是什么问题?对了,还是递归问题,根据以上分析我们不难写出如下的递归公式:

(作者原话)对公式的说明:(1)f(s,n)表示,有n个骰子,出现和为s的情况总数;(2)对于公式第二行的解释;如果有一个骰子,那么点数为8或者0的情况数,我们记为0,这样是为了在计算递归时更为方便所作的处理;例如有公式可知f(8,2)=f(7,1)+f(6,1)+f(5,1)+f(4,1)+f(3,1)+f(2,1),如果我们规定了f(7,1)=0那么计算会方便很多。

有了上面的分析,我们可以写出如下代码

var diceMaxValue:Int = 6

//计算给定diceNumber个骰子,出现和为diceTotalSum的所有可能情况的总数
//int fun(int diceTotalSum,int diceNumber)
func mainFun(diceTotalSum:Int,diceNumber:Int)->Int{
    //骰子数少于1,错误
    if diceNumber < 1 {
        return 0
    }
    //骰子数等于1,如果超出1~6范围便错误,未超出则是1
    if diceNumber == 1 {
        if (diceTotalSum >= diceNumber && diceTotalSum <= diceMaxValue*diceNumber) {
            return 1
        }
        else {
            return 0
        }
    }
    //n>1,采用递归
    if diceNumber > 1 {
        var sum:Int = 0
        for var index:Int = 1; index <= diceMaxValue; ++index {
            sum += mainFun(diceTotalSum-index,diceNumber-1)
        }
        return sum
    }
    return 0
}

//给定number个骰子,打印出现各种情况的概率
func printSumProbabilityOfDice1(number:Int) {
    if number < 1 {
        return
    }

    var maxSum:Int = number * diceMaxValue
    var pProbability:Array<Float> = Array()
    for index in number...maxSum {
        pProbability.insert(Float(mainFun(index,number)),atIndex: index-number)
    }
    var total:Int = Int(pow(Float(diceMaxValue),Float(number)))

    for count in 0...maxSum-number {
        pProbability@H_301_158@[count@H_301_158@] /= Float(total) println("\(count+number)\t\(pProbability@H_301_158@[count@H_301_158@])") } }

(作者原话)分析2:和算法2中求解斐波那契数列的方法类似,递归的效率太差,我们可以正向来求解,假设我们有一个数组表示k-1个骰子中各点数的情况,令第s个分量表示和为s时情况总数,那么当有k个骰子是,其和为s时的情况总数,就是表示k-1骰子的数组中的s-1,s-2,s-6的和(分别令引入的第k个骰子的值分别取1,2,3,4,5,6即可,其实和分析1的思路差不太多)。根据这个思想,我们可以用两个数组交替表示数组k-1和数组k,于是我们有如下的代码

func printSumProbabilityOfDice2(number:Int){
    //创建并初始化一个二维数组
    var pProbability:Array = Array(count: 2,repeatedValue: Array(count: number * diceMaxValue + 1,repeatedValue: Double(0)))
    var flag:Int = 0
    //将第一个数组下标为1~6的赋值为1
    for index in 1...diceMaxValue {
        pProbability[flag][index] = Double(1)
    }

    //骰子数k从2到n循环;对于每一k,s取值为[k,6k],对于每一个s计算前一个数组
    //的s-1,s-6;因为前一个数组的最小值为k-1,因为因而有s-j>=k-1;
    for k in 2...number {
        for s in k...diceMaxValue*k {
            pProbability[1-flag][s] = 0
            for var j = 1; j <= diceMaxValue && j <= s - k + 1; j++ {
                pProbability[1-flag][s] += pProbability[flag][s-j]
            }
        }
        flag = 1 - flag
    }

    var total:Double = pow(Double(diceMaxValue),Double(number))
    for ss in number...number*diceMaxValue {
        pProbability[flag][ss] /= total
        println("\t\(pProbability[flag][ss])")
    }
}

(作者原话)最后分析:我们看到和【算法02】中提到的一样,虽然该算法的时间复杂度提高了很多,但是动态创建了两个数组,而且每一个的数组长度都没分析1中的长度多了一个n,因而还是以空间换时间的思想。好了,这个算法就到这,祝各位愉快!

参考文献:

【1】何海涛博客http://zhedahht.blog.163.com/blog/static/254111742009101524946359/

【2】Wikipedia:http://en.wikipedia.org/wiki/Dice#Probability

【3】http://www.helium.com/items/1538174-how-to-calculate-probability-using-multiple-dice

原文链接:https://www.f2er.com/swift/327205.html

猜你在找的Swift相关文章