动态规划不同维度分析leetcode198.打家劫舍问题

java 复制代码

class Solution {
    public int rob(int[] nums) {
        return robByTwoDim(nums);
    }

    // 二维dp算法 一层for训练
    public int robByTwoDim(int[] nums){
        int[][] dp = new int[2][nums.length + 1];
        for(int j = 1; j <= nums.length; j++){
            dp[0][j] = nums[j - 1] + dp[1][j - 1];   // 偷,那么再去上一个不偷的最大值
            dp[1][j] = Math.max(dp[0][j - 1], dp[1][j - 1]);  // 不偷，在上一个偷和不偷之间选择一个
        }
        return Math.max(dp[0][nums.length], dp[1][nums.length]);
    }

    // 一维dp算法 两层for循环
    public int robByOneDim(int[] nums) {
        int[] dp = new int[nums.length + 1];
        int max = 0;
        for(int i = 1; i <= nums.length; i++){
            for(int j = 0; j < i - 1; j++){  // i - 1 保证是当前可以偷
                dp[i] = Math.max(dp[i], dp[j]);
            }
            dp[i] = dp[i] + nums[i - 1];
            if(dp[i] > max){
                max = dp[i];
            }
        }
        return max;
    }
}

该题分别利用一维和二维dp数组进行求解。一般来说，遇到递归时，先思考一维再思考二维，对于复杂的问题，可直接先对二维进行思考。一维一般注意点：（1）dp数组中当前索引对应存储空间存储的是从下标0到当前索引最优值，还是必须考虑当前索引的次优值，对于该题中第一种解法，当前房间必须偷的最优值。全局最优可以利用max临时变量来进行记录。（2）当计算递推公式无法利用某一个或者某几个推导时，可以考虑，从第一个元素开始进行遍历。针对该题，如果当前房间必偷，dp数组当前索引推导式可在从0-(i-1)最大值加当前房间现金数。二维一般要注意五要素，特别是遍历顺序，可以思考对dp[i][j]中j进行遍历，正如该题，也可以按照二维数组遍历顺序进行遍历，当前值由(i - m (||、&&) j - n)s等所得。最后，为了方便编码，可以为边界预留空间，比如，本解法中dp数组长度都加了1。