LeetCode算法题198:打家劫舍解析

你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。

给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。

示例1:

输入: [1,2,3,1]
输出: 4
解释: 偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。
     偷窃到的最高金额 = 1 + 3 = 4 。

示例2:

输入: [2,7,9,3,1]
输出: 12
解释: 偷窃 1 号房屋 (金额 = 2), 偷窃 3 号房屋 (金额 = 9),接着偷窃 5 号房屋 (金额 = 1)。
     偷窃到的最高金额 = 2 + 9 + 1 = 12 。

这个题一看就是要用动态规划,设置一个数组保存到当前为止可以偷到的最多的金额,更新当前最高金额的方法就是将当前数加前前金额与前一最大金额对比,哪个大就是当前可偷最大金额。这样说来也就不需要存所有的过程最大金额了,只需要存两个金额,前前一个最大金额和前一个最大金额就可以计算了。
C++源代码:

class Solution {
public:
    int rob(vector<int>& nums) {
        int n = nums.size();
        if (n==0) return 0;
        if (n==1) return nums[0];
        int ppre = nums[0];
        int pre = ppre>nums[1]?ppre:nums[1];
        for (int i=2;i<n;i++)
        {
            int now = ppre + nums[i];
            ppre = pre;
            pre = now>pre?now:pre;
        }
        return pre;
    }
};

python3源代码:

class Solution:
    def rob(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        n = len(nums)
        if n==0:
            return 0
        ppre = 0
        pre = nums[0]
        for i in range(1,n):
            now = ppre + nums[i]
            ppre = pre
            pre = max(now, pre)
        return pre      

猜你喜欢

转载自blog.csdn.net/x603560617/article/details/83958202