动态规划解强盗抢劫房间问题

题目：

强盗抢劫一排房间（房间数大于3），每个房间都有钱，不能抢劫两个相邻的房间，要求抢的钱最多。每个房间的钱数用数组表示。数组如：[2,7,9,3,1]

解：
对每个元素进行判断，并存储当前元素下最大和。
如果输入为空，返回0；如果输入个数为1，返回这个数；如果说如个数为2，返回最大值。
输入个数大于等于3，就需要运用动态规划。将当前数字和往前数两个（因为是隔一个取）的数字和相加，并和往前数一个的数字和进行对比，取最大。dp[i] = max(dp[i - 2] + nums[i], dp[i - 1]);

# -*- coding: utf-8 -*-
"""
Created on Tue Oct 25 10:46:44 2022

@author: Dell
"""
def rob(nums) :
    #判断房间个数
    size = len(nums)
    if size == 0:
        return 0
    if size == 1:
        return nums[0]
    dp=[0]
    for i in range(size-1):
        dp.append(0)
    #初始化dp数组
    dp[0] = nums[0]
    if nums[0] > nums[1]:
        dp[1] = nums[0]
    else:
        dp[1] = nums[1]

    i = 2
    #计算相邻两个房间的价值，取最大值
    for i in range(size):
        dp[i] = max(dp[i-2] + nums[i], dp[i-1])
    return dp[size-1]

nums = [2, 7, 9, 3, 1]
print('最多能偷',rob(nums))

运行结果：