题目链接: 198.打家劫舍
dp数组:i
个房间,可以偷到金钱最大为dp[i]
。需要强调的是dp[i]
并不等价于一定要偷i
房间,只是表示包含了i
个房间,是否偷了房间,是通过nums[i]
来表示的
递推公式:
i
房间:dp[i - 2] + nums[i]
i
房间:dp[i - 1]
dp[i] = max(dp[i - 1], dp[i - 2] + nums[i])
。初始化:dp[0] = nums[0],dp[1] = max(nums[0], nums[1])
class Solution {
public:
int rob(vector<int>& nums) {
if (nums.size() == 1) return nums[0];
vector<int> dp{nums[0], max(nums[0], nums[1])};
dp.resize(nums.size());//不是nums.size() + 1,是因为dp是按照nums对应求得的
for (int i = 2; i < nums.size(); ++i) {
dp[i] = max(dp[i - 1], dp[i - 2] + nums[i]);
}
return dp.back();
}
};
题目链接:213.打家劫舍II
本题为环形循环,表示首、尾不能同时选择,讨论选择首、尾的最大值即可。
class Solution {
int robZero(const vector<int>& nums, int start, int end) {
vector<int> dp{nums[start], max(nums[start], nums[start + 1])};
for (int i = start + 2; i < end; ++i) {
dp.push_back(max(dp[i - start - 2] + nums[i], dp[i - start - 1]));
}
return dp.back();
}
public:
int rob(vector<int>& nums) {
if (nums.size() == 1) return nums[0];
if (nums.size() == 2) return max(nums[0], nums[1]);
return max(robZero(nums, 0, nums.size() - 1), robZero(nums, 1, nums.size()));
}
};
题目链接:337.打家劫舍III
树形dp,本题和968.监控二叉树 类似
dp数组:
每个节点只有两个状态:偷、不偷,用dp[0]
表示不偷当前节点获得最大金钱,dp[1]
表示偷当前节点获得最大金钱。这就是整个二叉树的dp数组
:因为当前节点可以通过该节点的父节点得到,因此只需知道其父节点的状态,就可以得到当前节点状态。
dp
数组为dp{不偷, 偷}
初始化 :
对于叶子节点来说,由于没有子节点,因此初始化为{0, 0}
遍历顺序:由于是二叉树,最后递归到根节点,因此采用后序遍历
class Solution {
vector<int> postOrder(TreeNode* cur) {
if (!cur) return {0, 0};//初始化
auto left = postOrder(cur->left);
auto right = postOrder(cur->right);
//偷当前节点
int val1 = cur->val + left[0]/*不偷左子节点*/ + right[0]/*不偷右子节点*/;
//不偷当前节点
int val2 = max(left[0], left[1]) + max(right[0], right[1]);
return {val2, val1};
}
public:
int rob(TreeNode* root) {
return max(postOrder(root)[0]/*不偷根节点*/, postOrder(root)[1]/*偷根节点*/);
}
};