力扣刷题 动态规划:打家劫舍全系列
这个系列有好几道,一个一个来!
198. 打家劫舍
难度:简单
题目描述
解题思路
这是这个系列里最简单的一道啦,状态转移方程很明显。对于第i个房子,要么偷要么不偷,如果偷那金额就等于前两天的金额+当前房子的金额;如果不偷那就等于前一天的金额。每天从这两种选择中选出金额最大的,写成状态转移方程就是:
dp[i]=max(dp[i−2]+nums[i],dp[i−1])
代码也挺简短的,注意边界条件和数组下标,时间和空间复杂度都是O(N)
public int rob(int[] nums) {
int n = nums.length;
if(n == 0)
return 0;
int[] dp = new int[n+1];
dp[1] = nums[0]; //初始化第一天
//今天偷 dp[i-2]+nums[i] 不偷 dp[i-1]
for(int i = 2;i < n + 1;i++){
dp[i] = Math.max(dp[i-2] + nums[i-1],dp[i-1]);
}
return dp[n];
}
状态压缩 因为当前状态只和前一天和前两天的金额有关,所以可以只用两个变量来分别存储,把空间复杂度从O(N)压缩成O(1)
public int rob(int[] nums) {
int n = nums.length;
if(n == 0)
return 0;
int pre1 = 0,pre2 = 0,max = 0; //前一天,前两天
pre1 = nums[0]; //初始化第一天
//今天偷 pre2+nums[i] 不偷 pre1
for(int i = 1;i < n