今天是第 天刷leetcode,立个flag,打卡60天,如果做不到,完成一件评论区点赞最高的挑战。
算法挑战链接
198. 打家劫舍https://leetcode.cn/problems/house-robber/
第一想法
计算可以抢到的最大值,这种题目的第一感觉就是动态规划。
动态规划的五部曲走起:
- 确定dp数组(dp table)以及下标的含义
dp[i] 代表的含义是:在 0-i 可以抢的情况下,抢的最大值是dp[i]
- 确定递推公式
dp[i]的值可以从两个方面去考虑。一个是需要抢 i 位置上的,一个是不需要抢。
假设抢了 i 位置上的,那么i-1位置上的就不能再抢了:dp[i-2] + nums[i-1]
假设没有抢 i 位置上的,那么 dp[i - 1]
因此递推公式为:dp[i] = max( dp[i - 1], dp[i-2] + nums[i-1]
- dp的初始化
因为有个i-2,因此i是从2开始遍历的,nums的取值是i-1 所以我们需要初始化 dp[1].在只有一个可以抢的情况下,最大值就是这一个的值,因此 dp[1] = nums[0];
- 确定遍历顺序
dp[i] 是根据dp[i - 2] 和 dp[i - 1] 推导出来的,那么一定是从前到后遍历!
- 举例推导dp数组
数据就不推导了。
代码如下:
class Solution {
public static int rob(int[] nums) {
int[] dp = new int[nums.length + 1];
dp[1] = nums[0];
for (int i = 2; i <= nums.length; i++) {
dp[i] = Math.max(dp[i-1], dp[i-2]+nums[i-1]);
}
return dp[nums.length];
}
}
今日收获
简简单单挺好的