- 198.打家劫舍
解题过程出现的问题:初始化dp数组出现了问题 dp[1] = max(nums[0], nums[1])
class Solution {
public:
int rob(vector<int>& nums) {
// dp[i] 表示到当前屋舍最多偷到了多少
// dp[i] = dp[i - 2] + nums[i] 或者 dp[i - 1]
vector<int> dp(nums.size(), 0);
if(nums.size() == 1) return nums[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 - 2] + nums[i], dp[i - 1]);
}
return dp[nums.size() - 1];
}
};
- 213.打家劫舍II
连成一个环 相当于 首位不能同时选
class Solution {
public:
int rob(vector<int>& nums) {
int n = nums.size();
if(n == 1) return nums[0];
int res1 = robLine(nums, 0, n - 2);
int res2 = robLine(nums, 1, n - 1);
return max(res1, res2);
}
int robLine(vector<int>& nums, int start, int end)
{
if(start == end) return nums[start];
int n = end - start + 1;
vector<int> dp(nums.size(), 0);
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];
}
};
- 337.打家劫舍III (一刷)
需要对树形结构进行后序遍历
用dp[0]表示不取当前节点的最大值 dp[1]表示取当前节点的最大值
dp[0] = 左子树的最大值 + 右子树的最大值
dp[1] = 当前节点的val + 不取左子树根节点最大值 + 不取右子树根节点最大值
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
int rob(TreeNode* root) {
vector<int> dp = robTree(root);
return max(dp[0], dp[1]);
}
// dp[0]不偷 dp[1]偷
vector<int> robTree(TreeNode* cur) {
if(cur == NULL) return{0, 0};
vector<int> leftdp = robTree(cur->left);
vector<int> rightdp = robTree(cur->right);
int val0 = max(leftdp[0], leftdp[1]) + max(rightdp[0], rightdp[1]);
int val1 = cur->val + leftdp[0] + rightdp[0];
return {val0, val1};
}
};