代码随想录算法训练营Day45|198.打家劫舍I、213.打家劫舍II、337.打家劫舍III

2024-06-22 00:52

本文主要是介绍代码随想录算法训练营Day45|198.打家劫舍I、213.打家劫舍II、337.打家劫舍III,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

打家劫舍I

198. 打家劫舍 - 力扣(LeetCode)

动态规划

动态数组dp[i]表示在前i+1个房屋能偷到的最大金额。

由于相邻不能偷的原则,dp[i] = max(dp[i-2]+nums[i],dp[i-1]),即为若要偷当前房屋,则比较前一个房屋dp[i-1]和不偷前一个房屋,但偷当前房屋的金额dp[i-2]+nums[i]的较大值。

对dp数组的初始化,从dp数组的推导可以看出,需要初始化最开始的2个元素才能开始递推,

dp[0]即只有1个房子,则只能偷那一个房子,dp[0] = nums[0],dp[1]则考虑第0个房屋和第1个房屋中哪个金额大,哪个金额大偷哪个,则dp[1] = max(nums[0],nums[1])

依次从前往后递推

最后返回dp[nums.size()-1]即为能偷到的最大金额。

class Solution {
public:int rob(vector<int>& nums) {// 如果没有房屋,则无法偷窃,返回0if(nums.size()==0)return 0;// 如果只有一家房屋,则直接偷窃这家房屋,返回其金额if(nums.size()==1)return nums[0];// 创建一个动态规划数组dp,用于存储到当前房屋为止能够偷窃的最大金额vector<int>dp(nums.size(),0);// 初始化dp数组的前两个值// 第一个房屋的最大偷窃金额为nums[0]dp[0] = nums[0];// 第二个房屋的最大偷窃金额为nums[0]和nums[1]中的较大值dp[1] = max(nums[0], nums[1]);// 从第三个房屋开始,计算到当前房屋为止能够偷窃的最大金额// 对于每个房屋j,可以选择偷或者不偷:// 如果偷,则最大金额为dp[j-2](即前一个房屋之前能偷的最大金额)加上当前房屋的金额nums[j]// 如果不偷,则最大金额为dp[j-1](即前一个房屋能偷的最大金额)// 取两者的较大值作为当前房屋能偷的最大金额for(int j = 2; j < nums.size(); j++) {dp[j] = max(dp[j-2] + nums[j], dp[j-1]);}// 返回dp数组的最后一个元素,即偷窃完所有房屋能够得到的最大金额return dp[nums.size()-1];}
};

代码的时间复杂度为O(n),空间复杂度为O(n)。

打家劫舍II

213. 打家劫舍 II - 力扣(LeetCode)

参考思路 代码随想录 (programmercarl.com)

这里我开头有想过不选头或者不选尾,但没有更深入了解下去,参考下代码随想录里的思路,总共有三种情况,分别是1.不选头尾,2.不选头,3.不选尾,2、3包含了1的情况,所以只考虑2、3,创建两个dp数组,dp数组1不偷尾部的房间,dp数组2不偷头部的房间,然后取2个数组末尾元素的较大值,即能得到能获得的最大金额。

其余操作和打家劫舍I相同。

class Solution {
public:int rob(vector<int>& nums) {// 如果没有房屋,则没有钱可偷if(nums.size() == 0)return 0;// 如果只有一家房屋,则偷这一家的钱if(nums.size() == 1){return nums[0];}// 如果只有两家房屋,则偷其中金钱较多的那家if(nums.size() == 2){return max(nums[0], nums[1]);}// 创建两个新的数组,分别包含第一家和第二家到最后一家的所有房屋vector<int> nums1(nums.begin(), nums.end() - 1);vector<int> nums2(nums.begin() + 1, nums.end());// 创建两个动态规划数组,用于存储偷到每家房屋时能够获得的最大金额vector<int> dp1(nums1.size(), 0);vector<int> dp2(nums1.size(), 0);// 初始化第一家和第二家的情况dp1[0] = nums1[0];dp1[1] = max(nums1[0], nums1[1]);dp2[0] = nums2[0];dp2[1] = max(nums2[0], nums2[1]);// 使用动态规划计算偷到每家房屋时能够获得的最大金额,不偷相邻的房屋for(int i = 2; i < nums1.size(); i++) {dp1[i] = max(dp1[i - 2] + nums1[i], dp1[i - 1]);}for(int i = 2; i < nums2.size(); i++) {dp2[i] = max(dp2[i - 2] + nums2[i], dp2[i - 1]);}// 返回两种情况中能够获得的最大金额return max(dp1[nums1.size() - 1], dp2[nums2.size() - 1]);}
};

代码随想录里面的代码要简洁明了些,贴在下面。

class Solution {
public:int rob(vector<int>& nums) {if (nums.size() == 0) return 0;if (nums.size() == 1) return nums[0];int result1 = robRange(nums, 0, nums.size() - 2); // 情况二int result2 = robRange(nums, 1, nums.size() - 1); // 情况三return max(result1, result2);}// 198.打家劫舍的逻辑int robRange(vector<int>& nums, int start, int end) {if (end == start) return nums[start];vector<int> dp(nums.size());dp[start] = nums[start];dp[start + 1] = max(nums[start], nums[start + 1]);for (int i = start + 2; i <= end; i++) {dp[i] = max(dp[i - 2] + nums[i], dp[i - 1]);}return dp[end];}
};

算法的时间复杂度和空间复杂度均为O(n)。

打家劫舍III

337. 打家劫舍 III - 力扣(LeetCode)

递归

由于是二叉树,考虑递归的思路,对于当前节点,有抢和不抢两种选择,如果当前节点抢了,那么当前节点的子节点就不能抢了(题目要求),则获得的金额为root.val+rob(root->left->left)+rob(root->left->right)+rob(root->right->left)+rob(root->right->right),若当前节点不抢,则能抢子节点rob(root->left)+rob(root->right)。则我们需要选择得到这两种的较大值。但是如果从头到尾递归的时间复杂度过大,leetcode会超时,使用记忆化递归,使用一个map将计算过的结果保存。

代码随想录 (programmercarl.com)

class Solution {
public:unordered_map<TreeNode* , int> umap; // 记录计算过的结果int rob(TreeNode* root) {if (root == NULL) return 0;if (root->left == NULL && root->right == NULL) return root->val;if (umap[root]) return umap[root]; // 如果umap里已经有记录则直接返回// 偷父节点int val1 = root->val;if (root->left) val1 += rob(root->left->left) + rob(root->left->right); // 跳过root->leftif (root->right) val1 += rob(root->right->left) + rob(root->right->right); // 跳过root->right// 不偷父节点int val2 = rob(root->left) + rob(root->right); // 考虑root的左右孩子umap[root] = max(val1, val2); // umap记录一下结果return max(val1, val2);}
};

算法的时间复杂度为O(n),空间复杂度为O(logn)。

动态规划

看代码随想录,不太理解,周末再看看

代码随想录 (programmercarl.com)

树形动态规划的入门题。

每个节点仅有两个状态,偷或者不偷,以一个长度为2的一维dp数组即可表示,dp[0]表示不偷当前节点获得的最大金额,dp[1]表示偷当前节点所能获得的最大金额,当前层的dp数组表示了当前层偷或不偷的状态。

使用后续遍历,数值从底部向上传递。最后返回root的dp[0]和dp[1]的较大值。

偷当前节点的金额为val1 = cur->val + leftdp[0] + rightdp[0],dp[0]表示不偷

不偷当前节点的金额为 val2 = max(leftdp[0],leftdp[1])+max(rightdp[0],rightdp[1]),此处dp[0]和dp[1]是递归上来的子节点的金额值。

class Solution {
public:int rob(TreeNode* root) {vector<int> result = robTree(root);return max(result[0], result[1]);}// 长度为2的数组,0:不偷,1:偷vector<int> robTree(TreeNode* cur) {if (cur == NULL) return vector<int>{0, 0};vector<int> left = robTree(cur->left);vector<int> right = robTree(cur->right);// 偷cur,那么就不能偷左右节点。int val1 = cur->val + left[0] + right[0];// 不偷cur,那么可以偷也可以不偷左右节点,则取较大的情况int val2 = max(left[0], left[1]) + max(right[0], right[1]);return {val2, val1};}
};

时间复杂度为O(n),空间复杂度为O(logn)。

这篇关于代码随想录算法训练营Day45|198.打家劫舍I、213.打家劫舍II、337.打家劫舍III的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/1082826

相关文章

不懂推荐算法也能设计推荐系统

本文以商业化应用推荐为例,告诉我们不懂推荐算法的产品,也能从产品侧出发, 设计出一款不错的推荐系统。 相信很多新手产品,看到算法二字,多是懵圈的。 什么排序算法、最短路径等都是相对传统的算法(注:传统是指科班出身的产品都会接触过)。但对于推荐算法,多数产品对着网上搜到的资源,都会无从下手。特别当某些推荐算法 和 “AI”扯上关系后,更是加大了理解的难度。 但,不了解推荐算法,就无法做推荐系

康拓展开(hash算法中会用到)

康拓展开是一个全排列到一个自然数的双射(也就是某个全排列与某个自然数一一对应) 公式: X=a[n]*(n-1)!+a[n-1]*(n-2)!+...+a[i]*(i-1)!+...+a[1]*0! 其中,a[i]为整数,并且0<=a[i]<i,1<=i<=n。(a[i]在不同应用中的含义不同); 典型应用: 计算当前排列在所有由小到大全排列中的顺序,也就是说求当前排列是第

csu 1446 Problem J Modified LCS (扩展欧几里得算法的简单应用)

这是一道扩展欧几里得算法的简单应用题,这题是在湖南多校训练赛中队友ac的一道题,在比赛之后请教了队友,然后自己把它a掉 这也是自己独自做扩展欧几里得算法的题目 题意:把题意转变下就变成了:求d1*x - d2*y = f2 - f1的解,很明显用exgcd来解 下面介绍一下exgcd的一些知识点:求ax + by = c的解 一、首先求ax + by = gcd(a,b)的解 这个

综合安防管理平台LntonAIServer视频监控汇聚抖动检测算法优势

LntonAIServer视频质量诊断功能中的抖动检测是一个专门针对视频稳定性进行分析的功能。抖动通常是指视频帧之间的不必要运动,这种运动可能是由于摄像机的移动、传输中的错误或编解码问题导致的。抖动检测对于确保视频内容的平滑性和观看体验至关重要。 优势 1. 提高图像质量 - 清晰度提升:减少抖动,提高图像的清晰度和细节表现力,使得监控画面更加真实可信。 - 细节增强:在低光条件下,抖

【数据结构】——原来排序算法搞懂这些就行,轻松拿捏

前言:快速排序的实现最重要的是找基准值,下面让我们来了解如何实现找基准值 基准值的注释:在快排的过程中,每一次我们要取一个元素作为枢纽值,以这个数字来将序列划分为两部分。 在此我们采用三数取中法,也就是取左端、中间、右端三个数,然后进行排序,将中间数作为枢纽值。 快速排序实现主框架: //快速排序 void QuickSort(int* arr, int left, int rig

动态规划---打家劫舍

题目: 你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。 给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额。 思路: 动态规划五部曲: 1.确定dp数组及含义 dp数组是一维数组,dp[i]代表

活用c4d官方开发文档查询代码

当你问AI助手比如豆包,如何用python禁止掉xpresso标签时候,它会提示到 这时候要用到两个东西。https://developers.maxon.net/论坛搜索和开发文档 比如这里我就在官方找到正确的id描述 然后我就把参数标签换过来

poj 3974 and hdu 3068 最长回文串的O(n)解法(Manacher算法)

求一段字符串中的最长回文串。 因为数据量比较大,用原来的O(n^2)会爆。 小白上的O(n^2)解法代码:TLE啦~ #include<stdio.h>#include<string.h>const int Maxn = 1000000;char s[Maxn];int main(){char e[] = {"END"};while(scanf("%s", s) != EO

poj 1258 Agri-Net(最小生成树模板代码)

感觉用这题来当模板更适合。 题意就是给你邻接矩阵求最小生成树啦。~ prim代码:效率很高。172k...0ms。 #include<stdio.h>#include<algorithm>using namespace std;const int MaxN = 101;const int INF = 0x3f3f3f3f;int g[MaxN][MaxN];int n

秋招最新大模型算法面试,熬夜都要肝完它

💥大家在面试大模型LLM这个板块的时候,不知道面试完会不会复盘、总结,做笔记的习惯,这份大模型算法岗面试八股笔记也帮助不少人拿到过offer ✨对于面试大模型算法工程师会有一定的帮助,都附有完整答案,熬夜也要看完,祝大家一臂之力 这份《大模型算法工程师面试题》已经上传CSDN,还有完整版的大模型 AI 学习资料,朋友们如果需要可以微信扫描下方CSDN官方认证二维码免费领取【保证100%免费