LeetCode(C++)-动态规划(打家劫舍、打家劫舍II、打家劫舍III)

198. 打家劫舍

代码：

class Solution {	//198. 打家劫舍
	//动态规划五部曲：
	//1. 确定dp数组以及下标的含义：下标为i以内的房屋，能够偷窃到的最高金额为dp[i]
	//2. 确定递推公式：dp[i]=max(dp[i-2]+nums[i],dp[i-1]);
	//3. dp数组如何初始化：dp[0]=nums[0]  dp[1]=max(nums[0],nums[1])
	//4. 确定遍历顺序：从前往后
	//5. 举例推导dp数组：
public:
	int rob(vector<int>& nums) {
		if (nums.size() == 1)return nums[0];
		vector<int> 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];
	}
};

213. 打家劫舍 II

代码：

class Solution {	//213. 打家劫舍 II  环形的分两种情况，一是不算首元素，二是不算尾元素
	//动态规划五部曲：
	//1. 确定dp数组以及下标的含义：下标为i以内的房屋，能够偷窃到的最高金额为dp[i]
	//2. 确定递推公式：dp[i]=max(dp[i-2]+nums[i],dp[i-1]);
	//3. dp数组如何初始化：dp[start]=nums[start]  dp[start+1]=max(nums[start],nums[start+1])
	//4. 确定遍历顺序：从前往后
	//5. 举例推导dp数组：
public:
	int rob(vector<int>& nums) {
		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);
	}
	
	int robRange(vector<int>& nums, int start, int end) {
		if (start == end)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];
	}
};

337. 打家劫舍 III

代码：

class Solution {	//337. 打家劫舍 III  树形dp
	//动态规划五部曲：
	//1. 确定dp数组以及下标的含义：下标为0记录不偷该节点所得到的的最大金钱，下标为1记录偷该节点所得到的的最大金钱
	//2. 确定递推公式：  当前节点的最大金额=max（偷当前节点得到的最大金额，不偷当前节点得到的最大金额）
	//3. dp数组如何初始化：遇到空节点，返回｛0，0｝
	//4. 确定遍历顺序：后序遍历，通过递归左节点，得到偷与不偷的金钱，通过递归右节点，得到偷与不偷的金钱
	//5. 举例推导dp数组：
public:
	vector<int> robTree(TreeNode* cur) {
		if (cur == nullptr) return vector<int>{0, 0};
		vector<int> left = robTree(cur->left);
		vector<int> right = robTree(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 };
	}
	int rob(TreeNode* root) {
		vector<int> result = robTree(root);
		return max(result[0], result[1]);
	}
};

参考资料：

代码随想录