我用java刷 leetcode 198. 打家劫舍

Posted 2021-06-02 深林无鹿

这里有leetcode题集分类整理！！！

我的AC：

class Solution {
    public int rob(int[] nums) {
        int len = nums.length;
        if (len == 1) return nums[0];
        int[] dp = new int[len];
        dp[0] = nums[0];
        dp[1] = Math.max(nums[0], nums[1]);
        for (int i = 2 ; i < len ; i ++) {
            dp[i] = Math.max(dp[i - 2] + nums[i] , dp[i - 1]);
        }
        return dp[len - 1];
    }
}

滚动数组做空间优化：

class Solution {
    public int rob(int[] nums) {
        int len = nums.length;
        if (len == 1) return nums[0];
        int first = nums[0];
        int second = Math.max(nums[0], nums[1]);
        for (int i = 2 ; i < len ; i ++) {
            int tmp = Math.max(first + nums[i] , second);
            first = second;
            second = tmp;
        }
        return second;
    }
}