leetcode - 自顶向下解决 198.打家劫舍 问题

先确定递归结构
对于[1,2,3,1]的房子
我们可以有以下几种偷法
一: 1 +[3,1]
二:2+[1]
三:3
四:1

对于n个房子可以
一:1+[3,4,。。。,n]
二:2+【4,5,。。。,n】
三:3+【5,6,。。。。,n】



N:n

这样递归结构就出来了 res 随便赋一个初始值0
这是tryRob的函数的定义
也就是我们要递归的函数
int tryRob(int[] nums, int index)
递归的结束条件就是
if(index >= nums.length){ return 0; }
递归的方程:当前的值 + 剩余的房子总值(从下下个房子开始偷)

for(int i = index;i < nums.length;i++){
            res = Math.max(res, nums[i] + tryRob(nums, i+2));
        }

这样 我们完成了基本的递归结构 但是我们发现现在会有很多的重复计算,这就需要一个记忆化的数组 来存储算过的值
我们叫 memo[] 给他赋初值为-1 memo[i] 代表抢劫nums[i...n)所获得的最大收益

memo = new int[nums.length];
        for(int i = 0;i<nums.length;i++){
            memo[i] = -1;
        }

然后在哪里判断呢?我们在每次进入递归函数的时候就要进行判断

private int tryRob(int[] nums, int index){
        if(index >= nums.length){
            return 0;
        }
        if(memo[index] != -1){
            return memo[index];
        }

总体的代码如下:

public class Rob_02 {
    private int[] memo;
    public int rob(int[] nums) {
        memo = new int[nums.length];
        for(int i = 0;i<nums.length;i++){
            memo[i] = -1;
        }
        return tryRob(nums, 0);
    }
    private int tryRob(int[] nums, int index){
        if(index >= nums.length){
            return 0;
        }
        if(memo[index] != -1){
            return memo[index];
        }
        int res = 0;
        for(int i = index;i < nums.length;i++){
            res = Math.max(res, nums[i] + tryRob(nums, i+2));
        }
        memo[index] = res;
        return res;
    }
}
全部评论

相关推荐

dongsheng66:如果想进大厂的话,在校经历没必要占这么大篇幅,可以把专业技能单独放一个专栏写,可以加个项目经历
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务