复旦大学在读博士:动态规划使用简明总结

Python与算法社区

共 1380字,需浏览 3分钟

 ·

2020-11-25 22:16

Python与算法社区
441篇原创,干货满满
值得星标


01

02

03


三步加星标





你好,我是 zhenguo

今天很高兴为你推送一篇来自复旦大学在读博士 无尘 的关于动态规划的总结,话不多说,先睹为快。


从数学的视角来看,动态规划是一种运筹学方法,是在多轮决策过程中的最优方法。

那么,什么是多轮决策呢?其实多轮决策的每一轮都可以看作是一个子问题。从分治法的视角来看,每个子问题必须相互独立。但在多轮决策动态规划中,这个假设显然不成立,这是与分治法的明显不同之处,也是动态规划方法产生的原因之一。

动态规划是候选人参加面试的噩梦,也是面试过程中的难点。以下我根据自己的理解总结了动态规划相关的一些重要概念,便于从宏观上抽象理解这个算法。如果有不清楚的地方,非常欢迎大家留言讨论。

这是我绘制的动态规划知识点的一个思维导图:

下面举一个例子:

题目

有若干硬币排成一个序列,面值依次为[5,1,2,10,6,2],现在规定不能捡相邻的硬币,请问如果捡硬币才能使得捡到的硬币总价值最大。

思路

我们需要设计dp表来作为动态规划的数据模型,dp的值显然是硬币的总价值,表格的第一维是第几枚硬币,决策(也就是选择)有两个:

  • 第一个是选择第i-1个硬币
  • 第2个是不选择第i-1个硬币

这样我们就可以写出状态转移方程:

dp[i] = max(dp[i-2]+row_coins[i-1],dp[i-1])

注意我们还需要定义边界条件:

dp[0] = 0
dp[1] = row_coins[0]

这样我们就可以得到如下求解代码,最后编写traceback函数通过回溯来把选择的硬币打印出来。

代码


# 通过自底向上的方法来找到最优解
def bottom_up_coins(row_coins):
 dp = [None] * (len(row_coins)+1)
 dp[0] = 0
 dp[1] = row_coins[0]

 for i in range(2,len(row_coins)+1):
  dp[i] = max(dp[i-2]+row_coins[i-1],dp[i-1])

 return dp

c = [5,1,2,10,6,2]
dp = bottom_up_coins(c)

# 通过对dp的回溯来反向查找被选择的硬币
def trace_back_coins(c,dp):
 select = []
 i = len(c)
 while i>=1:
  if dp[i] > dp[i-1]:
   select.append(c[i-1])
   i -= 2
  else:
   i -= 1
 return select

print(dp)
print(trace_back_coins(c,dp))

运行结果

我们看到,用这种方法,很快能求得我们在选取 2、10、5这三枚硬币的时候,能得到最优的结果 17.

感谢无尘博士以上对动态规划的总结,若你意犹未尽,可以阅读学习为你准备的 Python 数据分析和算法 这本经典的电子书,微信备注:算法

不必打赏
给我点个赞
就心满意足了
浏览 21
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报