自学内容网 自学内容网

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

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。


原文地址:https://blog.csdn.net/shaui058/article/details/143824351

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!