欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

打家劫舍--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)
};
相关标签: JavaScript