213. 打家劫舍II

动态规划

class Solution {
    public int rob(int[] nums) {

        if (nums.length == 1){
            return nums[0];
        }

        int[] dp = new int[nums.length];
        int max = 0;

        /**
         * 重复两次《198. 打家劫舍》
         * 第一次不抢第一间房屋,第二次不抢最后一间房屋
         */
        dp[0] = 0;
        dp[1] = nums[1];

        for (int i = 2; i < nums.length; i++) {
            dp[i] = Math.max(dp[i - 2] + nums[i], dp[i - 1]);
        }

        max = dp[nums.length - 1];

        dp[0] = nums[0];
        dp[1] = Math.max(nums[0], nums[1]);

        for (int i = 2; i < nums.length - 1; i++) {
            dp[i] = Math.max(dp[i - 2] + nums[i], dp[i - 1]);
        }

        max = Math.max(max, dp[nums.length - 2]);

        return max;
    }
}

/**
 * 时间复杂度 O(n)
 * 空间复杂度 O(n)
 */

https://leetcode-cn.com/problems/house-robber-ii/

posted @ 2022-01-21 19:49  振袖秋枫问红叶  阅读(95)  评论(0)    收藏  举报