LeetCode 198.打家劫舍

动态规划问题，倒推公式

到第i户为止，有两种选择：

在第i-2偷盗利益最大化的基础上，再偷第i户

或者选择不偷本户，继续维持第i-1的偷盗利益

则有dp[i]=max(dp[i-1],dp[i-2]+A[i])（i>=2）

那么边界条件：dp[0]即第0户，dp[1]则是第0户和第1户之间的较大者

#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#include <vector>
using namespace std;

/*
* 思路：动态规划问题，假定偷盗顺序从左到右（不影响结果）
* 对于第i户来说，dp[i]=max(dp[i-1],dp[i-2]+A[i])
* 注意边界条件，只有两户时，到第二户为止的最优解是两户间的较大者
*/

class Solution {
public:
    int rob(vector<int>& nums) {
        int max_rob = nums[0], n = nums.size();
        int dp_p1 = nums[0], dp_p2;
        if (n >= 2)
        {
            // 注意这个边界条件
            dp_p2 = max(nums[0], nums[1]);
            max_rob = max(max_rob, dp_p2);
        }
        for (int i = 2; i < n; i++)
        {
            int p1 = dp_p1, p2 = dp_p2;
            dp_p2 = max(p2, p1 + nums[i]);
            dp_p1 = p2;
            max_rob = max(max_rob, dp_p2);
        }
        return max_rob;
    }
};


int main()
{
    int n, a;
    cin >> n;
    vector<int>nums;
    for (int i = 0; i < n; i++)
    {
        cin >> a;
        nums.push_back(a);
    }
    Solution A;
    cout << A.rob(nums) << endl;
}