动态规划系列之一爬楼梯问题

动态规划

官方解释:
动态规划(Dynamic Programming,DP)是运筹学的一个分支,是求解决策过程最优化的过程。20世纪50年代初,美国数学家贝尔曼(R.Bellman)等人在研究多阶段决策过程的优化问题时,提出了著名的最优化原理,从而创立了动态规划。动态规划的应用极其广泛,包括工程技术、经济、工业生产、军事以及自动化控制等领域,并在背包问题、生产经营问题、资金管理问题、资源分配问题、最短路径问题和复杂系统可靠性问题等中取得了显著的效果。

动态规划是程序进阶路上必须要经过的一个山坡,所以记录在学习动态规划过程中的方法和思路。

动态规划系列文章一共分为九个部分,每一个部分都是一个独立的例子,难易程度逐渐递增。这九个例子分别是:

  1. 动态规划系列之一爬楼梯问题
  2. 动态规划系列之二最大和子数组
  3. 动态规划系列之三乘积最大子数组
  4. 动态规划系列之四梅花桩问题
  5. 动态规划系列之五打家劫舍
  6. 动态规划系列之六01背包问题
  7. 动态规划系列之七完全背包问题
  8. 动态规划系列之八多重背包问题
  9. 动态规划系列之九找零钱

爬楼梯问题

不管问题有多难,我们首先来个望闻问切。以下面这个爬楼梯问题作为开胃菜。请听题:

有一座高度是10级台阶的楼梯,从下往上走,每跨一步只能向上1级或者2级台阶。要求用程序来求出一共有多少种走法。比如,每次走1级台阶,一共走10步,这是其中一种走法。我们可以简写成 1,1,1,1,1,1,1,1,1,1。

那么先不看答案,你觉得这一个问题该如何解决呢?从下往上一步一步的走,for循环尝试所有的办法吗?每一步的走法都会影响后面的走法,所以计算的量非常庞大。那么使用动态规划的解题思路如何处理呢?

动态规划问题的解决方式有几个固定的套路,个人总结为:

  1. 建立数学模型
  2. 状态转移方程
  3. 确立边界值

第一步 建立数学模型

假设走到第10级台阶有f(10)种方法,走到第9级台阶有f(9)种方法。记住这里f(10)就是走10级台阶走法的数量,这是动态规划的常见数学模型

第二步 状态转移方程

走到 10 级台阶时前面有两种可能,一种是从第8级台阶走上来,一次跨了2级台阶;或者从第9级台阶走上来,一次跨了1级台阶。所以说 走第10级台阶的总数 = 到第9级台阶的走法 + 到第8级台阶的走法

f(10) = f(9) + f(8)

同样道理,走到第9级台阶时肯定有两个可能,要么从第7级台阶过来,要么从第8级台阶过来。所有

f(9) = f(8) + f(7)
f(8) = f(7) + f(6)
f(7) = f(6) + f(5)
.....

总结状态转移方程就是:

f(n) = f(n-1) + f(n-2)

第三步 确立边界值

第1级台阶只有一种走法,就是1步跨上去,第2级台阶有两种走法,分别是每次跨一步,两次;一次跨两级台阶,一次;所以:

f(1) = 1
f(2) = 2

代码实现

有了以上的理论基础就可以写出代码来实现这个上楼梯问题。

def up(n):
    if n == 1:
        return 1
    if n == 2:
        return 2

    total = up(n-1)  + up(n-2)

    return total

result = up(10)
print('台阶的走法为:%d' % result)
台阶的走法为:89

换一个思路

以上的思路是一种由上向下的解法思路,从最后一个值不断向前推导,直到遇到边界情况。可以换一个思路,从下往上看。走第一步台阶只有1种方法,走第二部台阶有2种方法,第三步台阶可以从第1步跨2级上来,也可以是从第2步跨1级上来。所以:
已知: 边界情况 f(1) = 1, f(2) = 2 ,并且知道f(3) = f(1) + f(2)
求解:f(10)

def up(n):
    if n == 1:
        return 1

    if n == 2:
        return 2
    
    a = 1
    b = 2
    sum_num = 0
    for i in range(3,n+1):
        # 状态转移方程:f(n) = f(n-1) + f(n-2)
        sum_num = a + b
        a = b
        b = sum_num

    return sum_num 

总结

那么到这里就已经完成了走楼梯的解法,从代码量来看,核心代码不超过10行,是不是没有想象的那么难呢?是的,动态规划并不是想象的那么难,只要能领会思路,了解解题套路就畅通无阻啦。
另外可以看到动态规划问题既可以使用迭代解法,也可以使用循环解法。从常见的解题思路来看,循环的方法比较多。那么如果你已经理解了动态规划的建立数学模型 + 状态转移 + 确立边界值,说明本篇就完全掌握了。

posted @ 2020-10-04 16:31  金色旭光  阅读(1058)  评论(0编辑  收藏  举报