题解 | #打家劫舍(二)#

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param nums int整型vector 
     * @return int整型
     */
    int rob(vector<int>& nums) {
        // write code here
        int n = nums.size();
        vector<vector<int>> f(n, vector<int>(2, 0));
        f[0][0] = -1e8, f[0][1] = nums[0];
        for (int i = 1; i < n; i++) {
            f[i][0] = max(f[i - 1][0], f[i - 1][1]);
            f[i][1] = f[i - 1][0] + nums[i];
        }
        int a = f[n - 1][0];

        f = vector<vector<int>>(n, vector<int>(2, 0));
        f[0][0] = 0, f[0][1] = -1e8;
        for (int i = 1; i < n; i++) {
            f[i][0] = max(f[i - 1][0], f[i - 1][1]);
            f[i][1] = f[i - 1][0] + nums[i];
        }
        int b = max(f[n - 1][0], f[n - 1][1]);

        return max(a, b);
    }
};

• 思路：动态规划，先修题目《打家劫舍Ⅰ》
• 注意要遍历两次！！！
• 由于所有房屋是环形链表，因此需要考虑第一个房屋选和不选对最后一个房屋的影响，分类讨论即可：
• 1、选第一个房屋：则初始化选第一个房屋金额为nums[0]，不选第一个房屋金额为负无穷，然后计算一次最大值a，选了第一个房屋最后一个房屋则不能选
• 2、不选第一个房屋：则初始化选第一个房屋金额为负无穷，不选第一个房屋金额为0，然后计算一次最大值b
• 3、最后取a和b的最大值
• 时间复杂度：O(n)
• 空间复杂度：O(n)