leetcode 198 打家劫舍

简介

看到这种题目立即推, 动态规划.

但是我列出的不是特别好的公式, 提供自己的公式和官方的公式.
res 是 自己创建的动态规划数组. nums[i] 我这里是都会选择nums[i], 但是会递推前面两个.

        for(int i=2; i<nums.size(); i++){
            for(int j = i-2; j>=0 && j >= i -4; j--)
                res[i] = max(res[j] + nums[i], res[i]);
        }

code

class Solution {
public:
    int rob(vector<int>& nums) {
        vector<int> res(nums.size(), 0);
        if(nums.size() >= 1) res[0] = nums[0];
        if(nums.size() >= 2) res[1] = nums[1];
        for(int i=2; i<nums.size(); i++){
            for(int j = i-2; j>=0 && j >= i -4; j--)
                res[i] = max(res[j] + nums[i], res[i]);
        }
        int maxdata = 0;
        for(int i=0; i<res.size(); i++){
            if(res[i] > maxdata){
                maxdata = res[i];
            }
        }
        return maxdata;
    }
};

TIPS

感觉官方的差不多给我的感觉也差不多.

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