作者:力扣 (LeetCode) 链接:https://leetcode-cn.com/leetbook/read/top-interview-questions-easy/xnq4km/ 来源:力扣(LeetCode) 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。
给定一个代表每个房屋存放金额的非负整数数组,计算你 不触动警报装置的情况下 ,一夜之内能够偷窃到的最高金额。
示例 1:输入:[1,2,3,1]输出:4解释:偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。偷窃到的最高金额 = 1 + 3 = 4 。示例 2:输入:[2,7,9,3,1]输出:12解释:偷窃 1 号房屋 (金额 = 2), 偷窃 3 号房屋 (金额 = 9),接着偷窃 5 号房屋 (金额 = 1)。偷窃到的最高金额 = 2 + 9 + 1 = 12 。
思路:
设dp[i] 为偷前i + 1 间房子的最大非法所得,然后开始人脑求解。
针对nums = [2,7,9,3,1],有如下求解过程:
2 > 0 7 > 2 9 + 2 > 7 3 + 7 > 9 + 2
1 + 9 + 2 > 3 + 7
总结出前几项公式:
dp[2] = Math.max(dp[0] + nums[2], dp[1]);
dp[3] = Math.max(dp[1] + nums[3], dp[2]);
dp[4] = Math.max(dp[2] + nums[4], dp[3]);
Bingo,递推公式得出为dp[i] = Math.max(dp[i - 2] + nums[i], dp[i - 1])
/**
* @param {number[]} nums
* @return {number}
*/
var rob = function(nums) {
if (nums.length < 3) {
return Math.max(nums[0] || 0, nums[1] || 0);
}
const dp = [nums[0], Math.max(nums[0], nums[1])];
for (let i = 2; i < nums.length; i++) {
dp[i] = Math.max(dp[i - 2] + nums[i], dp[i - 1]);
}
return dp[nums.length - 1];
};
