動態規劃
- 思路:
- 尋找狀態轉移方程:
- 假設有 n 個房間;
- 如果偷第 n 個房間,那么第 n - 1 個房間不偷,之前的 n - 2 個房間偷竊到了 M(n - 2),總共可以偷竊到?M(n - 2) + N(n);
- 如果不偷第 n 個房間,那么 n - 1 個房間可以偷竊到 M(n - 1);
- 所以,選擇其中最大的情況是利益最大化,即 M(n) = max(M(n-2) + N(n), M(n-1));
- 邊界條件:
- 一個房間的時候 M(1) = N(1)
- 兩個房間的時候 M(2) = max(N(1), N(2))
- 綜上,完整代碼:
class Solution {
public:int rob(vector<int>& nums) {if (nums.empty()) {return 0;}int size = nums.size();if (size == 1) {return nums[0];}std::vector<int> dp = std::vector<int>(size, 0);dp[0] = nums[0];dp[1] = std::max(nums[0], nums[1]);for (int i = 2; i < size; ++i) {dp[i] = std::max(dp[i - 2] + nums[i], dp[i - 1]);}return dp[size - 1];}
};