打家劫舍python实现

问题:

# 你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上
# 被小偷闯入,系统会自动报警。
#
# 给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额。
#
#
#
# 示例 1:
#
#
# 输入:[1,2,3,1]
# 输出:4
# 解释:偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。
#   偷窃到的最高金额 = 1 + 3 = 4 。

可以采用递归和dp算法实现

def steal_money(li, index):
    # li:[1,2,3,1], index = max(index-1, (index-2)+index)
    if li is None or index < 0:
        return 0
    if index == 0:
        return li[0]
    return max(steal_money(li, index-1), steal_money(li, index-2)+li[index])

def steal_money_v2(li, length):
    if li is None or length==0:
        return 0
    if length == 1:
        return li[0]
    dp = [0] * length
    dp[0], dp[1] = li[0], max(li[1], li[0])
    for i in range(2, length):
        dp[i] = max(dp[i-1], dp[i-2]+li[i])
    return dp[length-1]

 

posted @ 2021-05-19 23:46  今夜无风  阅读(111)  评论(0编辑  收藏  举报