互联网站建设,昆明学校网站建设,WordPress注册界面文字,深圳福田区是富人区吗代码随想录算法训练营第四十八天| 198 打家劫舍 213 打家劫舍II 337 打家劫舍III
LeetCode 198 打家劫舍
题目: 198.打家劫舍
动规五部曲#xff1a;
确定dp数组以及下标的含义
dp[i]#xff1a;考虑下标i#xff08;包括i#xff09;以内的房屋#xff0c;最多可以偷…代码随想录算法训练营第四十八天| 198 打家劫舍 213 打家劫舍II 337 打家劫舍III
LeetCode 198 打家劫舍
题目: 198.打家劫舍
动规五部曲
确定dp数组以及下标的含义
dp[i]考虑下标i包括i以内的房屋最多可以偷窃的金额为dp[i]。
确定递推公式
决定dp[i]的因素就是第i房间偷还是不偷。
dp[i] max(dp[i - 2] nums[i], dp[i - 1])
dp数组如何初始化
dp[0] 是 nums[0]dp[1]是nums[0]和nums[1]的最大值
即dp[1] max(nums[0], nums[1])
vectorint dp(nums.size());
dp[0] nums[0];
dp[1] max(nums[0], nums[1]);确定遍历顺序
dp[i] 是根据dp[i - 2] 和 dp[i - 1] 推导出来的从前到后遍历
for (int i 2; i nums.size(); i) {dp[i] max(dp[i - 2] nums[i], dp[i - 1]);
}举例来推导dp数组
整体代码
class Solution {
public:int rob(vectorint nums) {if (nums.size() 0) return 0;if (nums.size() 1) return nums[0];vectorint dp(nums.size());dp[0] nums[0];dp[1] max(nums[0], nums[1]);for (int i 2; i nums.size(); i) {dp[i] max(dp[i - 2] nums[i], dp[i - 1]);}return dp[nums.size() - 1];}
};LeetCode 213 打家劫舍II
题目: 213.打家劫舍II
此题与上题区别是有成环的情况主要分为
情况一考虑不包含首尾元素情况二考虑包含首元素不包含尾元素情况三考虑包含尾元素不包含首元素
整体代码
class Solution {
public:int rob(vectorint 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(vectorint nums, int start, int end) {if (end start) return nums[start];vectorint 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];}
};LeetCode 337 打家劫舍III
题目: 337.打家劫舍III
本题换成了树首先想到遍历方式
本题一定要后序遍历需要通过递归函数的返回值来做下一步计算。
确定递归函数的参数和返回值
vectorint robTree(TreeNode* cur) 本题dp数组是一个长度为2的数组
确定终止条件
在遍历的过程中如果遇到空节点的话很明显无论偷还是不偷都是0返回
if (cur NULL) return vectorint{0, 0};确定遍历顺序
使用后序遍历。通过递归函数的返回值来做下一步计算。
通过递归左节点得到左节点偷与不偷的金钱。
通过递归右节点得到右节点偷与不偷的金钱。
// 下标0不偷下标1偷
vectorint left robTree(cur-left); // 左
vectorint right robTree(cur-right); // 右
// 中确定单层递归的逻辑
如果偷当前节点左右孩子不能偷val1 cur-val left[0] right[0];
如果不偷当前节点左右孩子可以偷选一个最大的val2 max(left[0], left[1]) max(right[0], right[1]);
当前节点的状态就是{val2, val1}; 即{不偷当前节点得到的最大金钱偷当前节点得到的最大金钱}
vectorint left robTree(cur-left); // 左
vectorint 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};举例推导dp数组
整体代码
class Solution {
public:int rob(TreeNode* root) {vectorint result robTree(root);return max(result[0], result[1]);}// 长度为2的数组0不偷1偷vectorint robTree(TreeNode* cur) {if (cur NULL) return vectorint{0, 0};vectorint left robTree(cur-left);vectorint 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};}
};
