动态规划之打家劫舍

大纲

最近有人询问我 LeetCode 「打家劫舍」系列问题(英文版叫 House Robber)怎么做,上网调研后发现,这一系列题目的点赞非常之高,是比较有代表性和技巧性的动态规划题目,今天就来聊聊这道题目。


以下三道题目,可在看完本文后,练练手
打家劫舍 I
打家劫舍 II
打家劫舍 III


打家劫舍I就是其他衍生题的根,所以我就以打家劫舍I为例题讲解

题目

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

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

示例 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 。

提示:

  • 0 <= nums.length <= 100
  • 0 <= nums[i] <= 400

思路

对于初学者,而言,看过题之后,肯定会有一点朦胧。为什么呢(ο´・д・)??

初步分析以后,能看到,一间房屋是否偷取,取决于 上一间房屋与上两间房屋!

到此为止,就更能感受到一种纠缠的关系,不过通过这种关系,就能顺理成章的发现递推关系。所以这就涉及到了动态规划。

有一定基础的人都知道,动态规划算是一种公式题型,对应的就会出来一套公式

我比较佩服的一个博主Carl,他把动归总结成了五部曲,如下:

  1. 确定下标含义
  2. 确定递推公式
  3. dp数组如何初始化
  4. 确定遍历顺序
  5. 举例推导dp数组
第一步:确定下标含义

我们设一个dp数组,vector dp(nums.size(),0);

其中dp[i]:考虑下标i(包括i)以内的房屋,最多可以偷窃的金额为dp[i]

第二步:确定递推公式

假设一下,现在处于第i个节点

如果取上一个节点的情况下,因为本节点不能偷,故:dp[i]=dp[i-1];

如果不取上一个节点,那就能偷本节点,故有price[i]:dp[i]=dp[i-2]+nums[i];

综合而言,就得出来:dp[i] = max(dp[i - 2] + nums[i], dp[i - 1]);

第二步:dp数组如何初始化

从递推公式dp[i] = max(dp[i - 2] + nums[i], dp[i - 1]);可以看出,递推公式的基础就是dp[0] 和 dp[1]

故:dp[0] 肯定时 dp[0] = nums[0]

而dp[1],为了取最大值 dp[1] = max(dp[0],dp[1]);

cpp 复制代码
vector<int> dp(nums.size());
dp[0] = nums[0];
dp[1] = max(nums[0], nums[1]);
第三步:确定遍历顺序

dp[i] 是根据dp[i - 2] 和 dp[i - 1] 推导出来的,那么一定是从前到后遍历

cpp 复制代码
for (int i = 2; i < nums.size(); i++) {
    dp[i] = max(dp[i - 2] + nums[i], dp[i - 1]);
}
第四步:举例推导dp数组

这里就不细讲喽,一般是在出错时,把dp数组打印出来,分析哪里错了。


cpp 复制代码
class Solution {
public:
    int rob(vector<int>& nums) {
        // 特殊情况
        if(nums.size()==0) return 0;
        if(nums.size()==1) return nums[0];

        // 初始化
        vector<int> dp(nums.size(),0);
        dp[0]=nums[0];
        dp[1]=max(nums[0],nums[1]);

        // 遍历方式
        for(int i=2; i<nums.size(); ++i){
            dp[i]=max(dp[i-1],dp[i-2]+nums[i]);
        }
        
        return dp[nums.size()-1];
    }
};

总结

其他大打家劫舍,大多是本题的衍伸,这说明本题恰恰是一道经典题

虽然本题简单,但这就不重视了吗???

最后在此,送坚持到这里的读者一句话:
简单题,用来培养方法; 难题,用来突破自我; 两者结合,方能突破至高; 当难题,难得你受不了时,恰恰是因为你没有重视简单题!

希望大家有所收获。

相关推荐
翀哥~43 分钟前
奇偶ASCII值判断
c++·算法·ascii
凯思软件2 小时前
Abaqus应用场景解析:从汽车碰撞到航空航天非线性分析
人工智能·算法·机器学习
不吃香菜?7 小时前
逻辑回归在信用卡欺诈检测中的实战应用
算法·机器学习·逻辑回归
Kay_Liang8 小时前
探究排序算法的奥秘(下):快速排序、归并排序、堆排序
java·数据结构·c++·python·算法·排序算法
禺垣8 小时前
AdaBoost算法的原理及Python实现
人工智能·python·算法·机器学习·数据挖掘·adaboost·集成学习
qq_508576098 小时前
混淆矩阵(Confusion Matrix)横坐标
人工智能·算法·机器学习
LSQ的测试日记8 小时前
机器学习_KNN算法
人工智能·算法·机器学习
炬火初现9 小时前
[leetcode]2302.统计得分小于k的子数组
算法·leetcode·职场和发展
鱼嘻9 小时前
数据结构------C语言经典题目(6)
linux·c语言·开发语言·数据结构·算法
fantasy_arch10 小时前
SVT-AV1源码分析-函数svt_aom_motion_estimation_kernel
人工智能·算法·av1