python实现动态规划

参考博客https://cloud.tencent.com/developer/article/1453737

学习动态规划的笔记,博客中的两个小demo都用python实现了一次。

首先从斐波那契数列入手,斐波那契数列可以看做一个简单的动态规划。求f(n) = f(n-1) + f(n-2),这个大学都学过,用一个递归就可以求出来了,而且其实也是比较直观的。比如f(100) 就可以知道事f(99) + f(98)然后一次类推一个个算出来,但是这样子有一个问题,就是复杂度极高,O(2**n),指数爆炸!我尝试在python下用这个求100的斐波那契数,就已经要算很久了。于是乎就有了改进,通过一个表记录子问题的答案。这样子 一下子复杂度就下降成线性的O(n),在python代码下100,甚至1000很快就算出来(Ps.在python种递归调用超过1000是不被允许的,需要修改配置参数。而且斐波那契1000也太大了吧),最后还有一点优化空间,就是上面的思路都是从上往下算,还可以从下网上算,这就是动态规划的思路了。可以看到斐波那契数列之和他前两个数有关,那就可以从f(1)开始计算,算道f(100),复杂度为O(1)。

有了上面例子的铺垫,开始看硬币问题。类似01背包,需要用尽可能少的硬币数目拼凑初目标金额。

终于把硬币问题的三种解法(递归/备忘录/dp)看完了,说实话递归是最容易绕进去的,由于实在存在太多的分支导致就算想要一步一步的按照逻辑走一遍马上就会乱掉。而当最后写出dp的解法后就会发现是如此的简单优雅,代码贴上去。这里虽然现在理解了,但是肯定不过一段时间遗忘的,或者换一个形式出现又不知道怎么解决了,总之还需多看啊。只要理解了动态规划的中心思想,代码是如此的简单优雅。

fib_result = {}
coin_result_dict = {}


import sys
sys.setrecursionlimit(9000000) #这里设置大一些
'''
递归暴力求解,O(n**2)效率低下
python对递归的调用次数有限制,超过1000次就会报错,可以修改参数配置
import sys
sys.setrecursionlimit(9000000) #这里设置大一些
'''
def fib(n):
    if n == 1 or n == 2:
        return 1
    return fib(n-1) + fib(n-2)


'''
备忘录方法,通过记录子问题的解,减少计算次数
复杂度O(n)
'''
def fib_helper(n):
    if n == 1 or n == 2:
        return 1
    return result(n-1) + result(n-2)


def result(n):
    if not n in fib_result:
        fib_result[n] = fib_helper(n)
    return fib_result[n]


'''
动态规划的思路,自下而上的求解
'''
def fib_dp(n):
    dp_map = {1:1,2:1}
    for i in range(3, n+1):
        dp_map[i] = dp_map[i - 1] + dp_map[i - 2]
    return dp_map[n]


'''
进一步优化,只保存前两个信息,复杂度为O(1)
'''
def fib_dp2(n):
    if n < 2:
        return n
    prev = 0
    curr = 1
    for i in range(1, n):
        sum = prev + curr
        prev = curr
        curr = sum
    return curr


###################################################################################
'''
上面是斐波那契数列
下面的例子是硬币组合
即又c种不同面额的货币,c1,c2,c3...ck 需要组合成n元,最少需要多少个硬币
递归问题的关键是状态转移方程,在这个问题中硬币数目f(n) = 1 + min{f(n - ci)|i 属于 [i,k]}
'''


'''
先用递归的方法求解
'''
def coin_cur(c,n):
    ans = sys.maxsize
    if n == 0:
        return 0
    for c_single in c:
        if c_single > n:
            continue
        rest_amount = coin_cur(c, n - c_single)
        if rest_amount == -1:
            continue
        print(c_single)  # 输出一个硬币队列,结合函数输出,例如3,就是最后三个数组即是所需的硬币组合
        ans = min(ans, rest_amount + 1)  # 注意上面的状态转移方程,此处就是对比ci1情况下和ci2情况下那种需要用到的硬币少
    return -1 if ans == sys.maxsize else ans


'''
带备忘录的方法
'''
def coin_cur_help(c, n):
    ans = sys.maxsize
    if n == 0:
        return 0
    for c_single in c:
        if c_single > n:
            continue
        rest_amount = coin_result(c, n - c_single)
        if rest_amount == -1:
            continue
        else:
            #print(c_single)
            ans = min(ans,rest_amount + 1)
    return -1 if ans == sys.maxsize else ans


def coin_result(c, n):
    if not n in coin_result_dict:
        coin_result_dict[n] = coin_cur_help(c, n)
    return coin_result_dict[n]


'''
动态规划
'''
def coin_dp(c, n):
    dp_map = {0:0}
    for i in range(1, n+1):
        dp_map[i] = sys.maxsize
    for k in dp_map:
       for c_single in c:
           if k < c_single:
               continue
           else:
                dp_map[k] = min(dp_map[k], 1+dp_map[k - c_single])
    return dp_map[n] if dp_map[n] != sys.maxsize else -1


if __name__ == '__main__':
    c = [5,10]
    n = 7
    print(coin_dp(c, n))


©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,362评论 5 477
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,330评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,247评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,560评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,580评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,569评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,929评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,587评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,840评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,596评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,678评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,366评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,945评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,929评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,165评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 43,271评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,403评论 2 342

推荐阅读更多精彩内容