题解 | #打家劫舍(二)#

打家劫舍(二)

https://www.nowcoder.com/practice/a5c127769dd74a63ada7bff37d9c5815

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param nums int整型一维数组 
     * @return int整型
     */
    public int rob (int[] nums) {
        // write code here
        if(nums.length==1){
            return nums[0];
        }
        int dp1[]=new int[nums.length+1];
        int dp2[]=new int[nums.length+1];
        first(nums,dp1);
        twice(nums,dp2);
	  //找到最大值
        int max=Math.max(dp1[nums.length-1],dp2[nums.length]);
        return max;
    }
    //偷第一家,最后一家不偷
    public void first(int[] nums,int dp1[]){
        dp1[1]=nums[0];
        for(int i=2;i<=nums.length;i++){
            dp1[i]=Math.max(dp1[i-1],dp1[i-2]+nums[i-1]);
        }
    }
    //偷第二家,最后一家偷
    public void twice(int[] nums,int dp2[]){
        dp2[1]=0;
        for(int i=2;i<=nums.length;i++){
            dp2[i]=Math.max(dp2[i-1],dp2[i-2]+nums[i-1]);
        }
    }
}

全部评论

相关推荐

点赞 评论 收藏
分享
05-24 14:12
门头沟学院 Java
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务