力扣算法学习day31-3

力扣算法学习day31-3

213-打家劫舍 II

题目

代码实现

class Solution {
    public int rob(int[] nums) {
        if(nums.length == 1){
            return nums[0];
        }
        // dp
        // 思路：本题是在打家劫舍I的特殊版本，因为环形结构所以如果取了第一个就不能取最后一个，
        // 如果取了最后一个就不能取第一个，然后这两种情况的话取最大值进行比较即可。
        int result1 = robResult(0,nums.length-1,nums);
        int result2 = robResult(1,nums.length,nums);

        return Math.max(result1,result2);
    }
    public int robResult(int index1,int index2,int[] nums){
        if(index2 - index1 == 1){
            return nums[index1];
        }
        // 1.创建dp数组，dp[i]表示有i个房间最多能偷多少现金。
        int[] dp = new int[index2 - index1];

        // 2.确定迭代公式。dp[i] = Math.max(dp[i-1],dp[i-2] + nums[i])
        // 3.初始化
        dp[0] = nums[index1];
        dp[1] = Math.max(nums[index1],nums[index1+1]);

        // 4.确定遍历顺序
        for(int i = 2;i < dp.length;i++){
            dp[i] = Math.max(dp[i-1],dp[i-2] + nums[index1+2]);
            index1++;
        }

        return dp[dp.length-1];
    }
}