代码随想录刷题|LeetCode 198.打家劫舍 213.打家劫舍II 337.打家劫舍III

简介: 代码随想录刷题|LeetCode 198.打家劫舍 213.打家劫舍II 337.打家劫舍III

198.打家劫舍

题目链接:力扣

思路


打家劫舍是动态规划的经典题目,考虑题目后会发现:当前房屋偷与不偷取决于 前一个房屋和前两个房屋是否被偷了。当前状态和前面的状态是一种依赖关系


       要求的其实是从头开始偷,偷到当前家最大值是多少


1、确定dp数组含义


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


2、确定递推公式


       是否偷当前家有两种情况:


如果偷第i个房间,就说明第 i-1 个不能偷,就不用考虑dp[i-1],那么当前能偷盗的最大值为dp[i-2] + nums[i]

如果不偷第i个房间,就说明我们是要考虑 i - 1 房,那么当前能偷盗的最大值就是dp[i - 1]  

3、初始化dp数组


       根据递推公式可以看成,需要初始化的有dp[0] 和 dp[1]


       dp[0] = nums[0];

       dp[1] = Math.max(dp[0],dp[1]);


4、遍历方式


       后面的数据由前面得出,所以从前向后进行遍历


打家劫舍

class Solution {
    public int rob(int[] nums) {
        if (nums.length == 1) {
            return nums[0];
        }
        // 创建dp[]数组
        // dp[i]代表偷盗到 i 家总金额为dp[i]
        int[] dp = new int[nums.length];
        // 初始化dp数组
        dp[0] = nums[0];
        dp[1] = Math.max(nums[0],nums[1]);
        // 遍历更新dp数组
        for (int i = 2; i < nums.length; i++) {
            dp[i] = Math.max(dp[i-1],dp[i-2]+nums[i]);
        }
        return dp[nums.length-1];
    }
}


213.打家劫舍II

题目链接:力扣

思路


 这道题目和上一道题目的区别是,这道题目中的房屋形成了环,这种就会造成头尾这两件房屋是连接的,在上一道题目中,使用递推公式就可以避免偷连续的两家。这里怎么避免头尾两家重复被偷呢,那就是分情况尽心分析


情况一:包含首元素,但是不包含尾元素

情况二:包含尾元素,但是不包含首元素


       这两种情况中,都可以避免首尾同时被偷,这两种情况又是单纯的打家劫舍问题,计算出两个结果再进行比较就可以


打家劫舍II

class Solution {
    public int rob(int[] nums) {
        if (nums.length == 1) {return nums[0];}
        // 情况二,不考虑尾部元素
        int result1 = robRange(nums,0,nums.length-2);
        // 情况三,不考虑头部元素
        int result2 = robRange(nums,1,nums.length-1); 
        return Math.max(result1,result2);
    }
    public int robRange(int[] nums, int start, int end){
        if (start == end) return nums[start];
        // 创建dp数组
        int[] dp = new int[nums.length];
        // 初始化dp数组
        dp[start] = nums[start];
        dp[start + 1] = Math.max(nums[start],nums[start+1]);
        // 遍历填充dp数组
        for (int i = start + 2; i <= end; i++) {
            dp[i] = Math.max(dp[i-1],dp[i-2]+nums[i]);
        }
        return dp[end];
    }    
}


337.打家劫舍III

题目链接:力扣

思路


这是一道树形dp的入门题目,目前对这种题目还是束手无策,直接根据题解来复盘


       之前的题目都是再一维或者二维数组上进行状态转移,而在二叉树上就是对二叉树上的节点进行状态转移


1、确定递归函数的参数和返回值

要求的是一个节点偷与不偷的两个状态所得到的金钱,那么返回值就是一个长度为2的数组


int[] robTree(TreeNode root);


这个返回值就是 dp 数组

dp的含义是:下标为0记录不偷该节点所得到的最大金钱,下标为1记偷该节点得到的最大金钱

2、确定终止条件

在遍历遇到空节点的时候,就可以进行返回了


if (root == null) return int[]{0,0};

这也相当于dp数组的初始化

3、确定遍历顺序

首先知道的是使用后序遍历,因为要通过递归函数的返回值来做下一步计算

通过递归左节点,得到左节点偷与不偷的金钱

通过递归右节点,得到右节点偷与不偷的金钱

int[] left = robTree(root.left);  //左节点偷与不偷的情况
int[] right = robTree(root.right); // 右节点偷与不偷的情况

4、确定单层递归的逻辑

针对某个节点来进行分析:


如果偷当前节点,那么左右孩子就不能偷,那么 val1 = root.val + left[0] + right[0] (数组的下标0代表不偷当前节点的情况)

如果不偷当前节点,那么左右孩子就可以偷,而且应该偷两个中最大的那个,那么 val2 = max( left[0] , left[1] ) + max( right[0] , right [1] );


int val1 = root->val + left[0] + right[0];
// 不偷cur
int val2 = Math.max(left[0], left[1]) + max(right[0], right[1]);
return {val2, val1};

5、推导dp数组

打家劫舍III

class Solution {
    public int rob(TreeNode root) {
        int[] res = robAction(root);
        return Math.max(res[0],res[1]);
    }
    int[] robAction(TreeNode root) {
        int[] res = new int[2];
        if (root == null) {
            return res;
        }
        int[] left = robAction(root.left);
        int[] right = robAction(root.right);
        // 不偷当前节点的情况
        res[0] = Math.max(left[0],left[1]) + Math.max(right[0],right[1]);
        // 偷当前节点的情况
        res[1] = root.val + left[0] + right[0];
        return res;
    }
}
相关文章
|
2月前
|
Unix Shell Linux
LeetCode刷题 Shell编程四则 | 194. 转置文件 192. 统计词频 193. 有效电话号码 195. 第十行
本文提供了几个Linux shell脚本编程问题的解决方案,包括转置文件内容、统计词频、验证有效电话号码和提取文件的第十行,每个问题都给出了至少一种实现方法。
LeetCode刷题 Shell编程四则 | 194. 转置文件 192. 统计词频 193. 有效电话号码 195. 第十行
|
3月前
|
搜索推荐 索引 Python
【Leetcode刷题Python】牛客. 数组中未出现的最小正整数
本文介绍了牛客网题目"数组中未出现的最小正整数"的解法,提供了一种满足O(n)时间复杂度和O(1)空间复杂度要求的原地排序算法,并给出了Python实现代码。
114 2
|
22天前
|
机器学习/深度学习 人工智能 自然语言处理
280页PDF,全方位评估OpenAI o1,Leetcode刷题准确率竟这么高
【10月更文挑战第24天】近年来,OpenAI的o1模型在大型语言模型(LLMs)中脱颖而出,展现出卓越的推理能力和知识整合能力。基于Transformer架构,o1模型采用了链式思维和强化学习等先进技术,显著提升了其在编程竞赛、医学影像报告生成、数学问题解决、自然语言推理和芯片设计等领域的表现。本文将全面评估o1模型的性能及其对AI研究和应用的潜在影响。
17 1
|
2月前
|
数据采集 负载均衡 安全
LeetCode刷题 多线程编程九则 | 1188. 设计有限阻塞队列 1242. 多线程网页爬虫 1279. 红绿灯路口
本文提供了多个多线程编程问题的解决方案,包括设计有限阻塞队列、多线程网页爬虫、红绿灯路口等,每个问题都给出了至少一种实现方法,涵盖了互斥锁、条件变量、信号量等线程同步机制的使用。
LeetCode刷题 多线程编程九则 | 1188. 设计有限阻塞队列 1242. 多线程网页爬虫 1279. 红绿灯路口
|
3月前
|
存储 算法 Java
LeetCode经典算法题:打家劫舍java详解
LeetCode经典算法题:打家劫舍java详解
70 2
|
3月前
|
算法 Python
【Leetcode刷题Python】 LeetCode 2038. 如果相邻两个颜色均相同则删除当前颜色
本文介绍了LeetCode 2038题的解法,题目要求在一个由'A'和'B'组成的字符串中,按照特定规则轮流删除颜色片段,判断Alice是否能够获胜,并提供了Python的实现代码。
51 3
|
3月前
|
Python
【Leetcode刷题Python】50. Pow(x, n)
本文介绍了LeetCode第50题"Pow(x, n)"的解法,题目要求实现计算x的n次幂的函数,文章提供了递归分治法的详细解析和Python实现代码。
27 1
|
3月前
|
Python
【Leetcode刷题Python】LeetCode 478. 在圆内随机生成点
本文介绍了LeetCode 478题的解法,题目要求在给定圆的半径和圆心位置的情况下实现在圆内均匀随机生成点的功能,并提供了Python的实现代码。
30 1
|
3月前
|
算法 Python
【Leetcode刷题Python】295. 数据流的中位数
本文介绍了一种使用Python实现的数据结构,用以支持数据流中添加整数并返回当前所有元素的中位数,通过排序列表来计算中位数。
25 1
|
3月前
|
算法 Python
【Leetcode刷题Python】73. 矩阵置零
本文介绍了LeetCode第73题的解法,题目要求在给定矩阵中将所有值为0的元素所在的行和列全部置为0,并提供了一种原地算法的Python实现。
32 0
【Leetcode刷题Python】73. 矩阵置零