题解 | 打家劫舍(一)

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param nums int整型一维数组 
     * @return int整型
     */
    public int rob (int[] nums) {
        // write code here
        if (nums.length <= 3) {
            int tmp1 = 0, tmp2 = 0;
            if (nums.length >= 1) {
                tmp1 += nums[0];
            }
            if (nums.length >= 2) {
                tmp2 += nums[1];
            }
            if (nums.length >= 3) {
                tmp1 += nums[2];
            }
            return tmp1 > tmp2 ? tmp1 : tmp2;
        }
        int[] dp = new int[nums.length];
        dp[0] = nums[0];
        dp[1] = nums[1];
        dp[2] = nums[0] + nums[2];
        for (int i = 3; i < nums.length; i++) {
            dp[i] = Math.max(dp[i - 2], dp[i - 3]) + nums[i];
        }
        return Arrays.stream(dp).max().getAsInt();
    }
}