动态规划之打家劫舍

大纲

最近有人询问我 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 <= numsi <= 400

思路

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

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

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

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

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

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

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

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

第二步:确定递推公式

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

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

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

综合而言,就得出来:dpi = max(dpi - 2 + numsi, dpi - 1);

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

从递推公式dpi = max(dpi - 2 + numsi, dpi - 1);可以看出,递推公式的基础就是dp0 和 dp1

故:dp0 肯定时 dp0 = nums0

而dp1,为了取最大值 dp1 = max(dp0,dp1);

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

dpi 是根据dpi - 2 和 dpi - 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];
    }
};

总结

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

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

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

希望大家有所收获。

相关推荐
通信小呆呆36 分钟前
当算法有了“五感”:多模态数据融合如何向人体感官协同学习?
人工智能·学习·算法·机器学习·机器人
benben0441 小时前
强化学习之DQN算法族(基于gymnasium开发)
算法
何以解忧,唯有..2 小时前
Go语言循环语句详解:for、range与循环控制
开发语言·算法·golang
想吃火锅10053 小时前
【leetcode】88.合并两个有序数组js
算法
生成论实验室4 小时前
机器人:一个自主运动的系统
人工智能·算法·语言模型·机器人·自动驾驶·agi·安全架构
Qres8214 小时前
算法复键——树状数组
数据结构·算法
H178535090964 小时前
SolidWorks第四部分_直接实体建模特征9_替换面原理
线性代数·算法·机器学习·3d建模·solidworks
不会就选b4 小时前
算法日常・每日刷题--<二分查找>3
算法
绿算技术4 小时前
Mooncake 与绿算ForinnBase GroundPool如何联手打破推理僵局?
科技·算法·架构