[213. 打家劫舍 II] 动态规划思路

最新推荐文章于 2024-10-09 22:28:58 发布

dry86

最新推荐文章于 2024-10-09 22:28:58 发布

阅读量341

点赞数 9

分类专栏：算法 C++ 文章标签：动态规划算法 c++

本文链接：https://blog.csdn.net/weixin_43699108/article/details/135895853

版权

C++ 同时被 2 个专栏收录

12 篇文章 0 订阅

订阅专栏

算法

11 篇文章 0 订阅

订阅专栏

本文介绍了一道编程问题，如何在有环形结构的房间打劫中找到最大收益。通过动态规划的方法，分别计算两种情况下的最大钱数，最后返回两者中的较大值作为解。

摘要由CSDN通过智能技术生成

Problem: 213. 打家劫舍 II

思路

题目在打家劫舍的条件基础上增加了前和尾房间不能偷，相当于房间之间成环，但是还是要破环，找到打劫开始的位置，可能的情况：

如果从首房间(0号房间)开始打劫，那么最后一个房间(尾房间，n-1号房间)不能打劫；相似地，(尾房间不打劫)
如果从1号房间开始打劫，那么最后一个房间能打劫到；(首房间不打劫)
从1号房间开始打劫，打劫到n-2号房间(首尾房间都不打劫)

第3种情况可能会包含在第2种情况下，所以只需要比较第1和2种情况的打劫最大钱数，最大的即为解。

Code

class Solution {
public:
    int rob(vector<int>& nums) {

        int n = nums.size();
        if(n == 1)  return  nums[0];
        if(n == 2)  return  max(nums[0], nums[1]);
        vector<int> dp(n, 0);   // 下标表示当前能拿到最大的值
        dp[0] = nums[0];
        dp[1] = max(nums[0], nums[1]);
        
        for(int i = 2; i < n-1; ++i)    // 第1种情况
        {
            dp[i] = max( dp[i-1], dp[i-2]+nums[i] );    
        }

        vector<int> dp2(n, 0);   // 下标表示当前能拿到最大的值
        dp2[1] = nums[1];
        dp2[2] = max(nums[1], nums[2]);
        for(int i = 3; i < n; ++i)  // 第2种情况
        {
            dp2[i] = max( dp2[i-1], dp2[i-2]+nums[i] ); 
        }
        return max(dp[n-2], dp2[n-1]);  // 比较第1和2种情况的打劫最大钱数
    }
};