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

打家劫舍(一)

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

import java.util.*;



public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param nums int整型一维数组 
     * @return int整型
     */
     //备忘录
     private int[] memo;
    public int rob (int[] nums) {
        memo = new int[nums.length];
        Arrays.fill(memo,-1);
        //从第一家开始抢
        return dp(nums,0);
    }
    
    int dp(int[] nums,int start){
        //无人可抢了
        if(start>=nums.length){
            return 0;
        }
        if(memo[start]!=-1)
        return memo[start];
        //dp(nums,start+1)不抢当前家,去下一家
        //dp(nums,start+2)+nums[start]抢当前家去下下家
        int res = Math.max(dp(nums,start+1),dp(nums,start+2)+nums[start]);
        memo[start] = res;
        return res;
    }
}

必须加备忘录剪枝优化,不然会超时。

全部评论

相关推荐

点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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