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; } }