打家劫舍--dp问题
程序员文章站
2022-07-12 09:09:51
...
/**
* @param {number[]} nums
* @return {number}
*/
/**
* @param {number[]} nums
* @return {number}
*/
/*
*假设打劫前i家能偷到的最大钱数为f(i),第i家有的钱数为nums[i],则有以下公式:
当i为1时:f(i) = nums[i]
当i为2时:f(i) = max(nums[i],nums[i-1])
当i大于2时:f(i) = max(nums[i-1] , nums[i-2] + nums[i])
*/
var rob = function(nums) {
let n = nums.length;
let dp = new Array(n).fill(0)
if(n===0){
return 0
}
for(let i = 0;i<n;i++){
if(i===0){
dp[i]= nums[i]
}
else if(i===1){
dp[i]= Math.max(nums[i],nums[i-1])
}else{
dp[i]=Math.max(dp[i-1],dp[i-2]+nums[i])
}
}
/* dp数组里面存储的是打劫前i所能偷到的最大钱数
输出打劫了那些房子,原理是:首先找到打劫的最大钱数,找到第一个出现该钱数的位置
那么就表明之后没有再偷其他的,则该位置就是偷的最后一家;然后用打劫到的最大钱数减去
该家所拥有的钱数就表明在前i家应打劫的最大钱数maxMoney,也就表示第一个出现maxMoney
的位置就是倒数第二个偷的位置,以此类推直到maxMoney和第i家所拥有的钱数相等就证明
只打劫了这一家,因此该家是最后一家.*/
let maxMoney = Math.max(...dp)
for(let i = dp.length -1;i>=0;i--){
if(dp[i-1]<maxMoney){
console.log("偷了第"+(i+1)+"间房子");
maxMoney -= nums[i]
}
if(maxMoney===nums[i]){
console.log("偷了第"+(i+1)+"间房子");
}
}
return Math.max(...dp)
};
上一篇: 计蒜客 逃生+动态规划
下一篇: 动态DP