【中等】【213】打家劫舍II

【中等】【213】打家劫舍II


链接点我

题目描述

在这里插入图片描述

思路

和 [198]打家劫舍 的区别就是本题形成了一个环。
也就是第一家最后一家不能同时打劫。
可以把环状转化成两个单排排列问题。

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

下面这种写法是对一个数组整个求dp[i]

    public int rob(int[] nums) {
        if(nums == null|| nums.length == 0){
            return 0;
        }
        if(nums.length == 1){
            return nums[0];
        }
        return Math.max(robSingle(Arrays.copyOfRange(nums,1,nums.length)),robSingle(Arrays.copyOfRange(nums,0,nums.length-1)));
    }
    
    private int robSingle(int[] nums){
        if(nums == null || nums.length == 0){
            return 0;
        }
        if(nums.length == 1){
            return nums[0];
        }
        int[] dp = new int[nums.length];
        dp[0] = nums[0];
        dp[1] = Math.max(nums[0],nums[1]);
        for(int i = 2; i < nums.length ;i++){
            dp[i] = Math.max(dp[i-1],dp[i-2]+nums[i]);
        }
        return dp[nums.length-1];
    }
发布了55 篇原创文章 · 获赞 1 · 访问量 869

猜你喜欢

转载自blog.csdn.net/weixin_42469108/article/details/105069219