动态规划 ----算法

打家劫舍

你是一个专业的小偷,计划偷窃沿街的房屋。每间房内都藏有一定的现金,影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警。

给定一个代表每个房屋存放金额的非负整数数组,计算你在不触动警报装置的情况下,能够偷窃到的最高金额。

示例 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 。
动态规划的一个很重要的过程就是找到[状态] 和[状态转移方程] 在这个问题里 设i是当前屋子的下标 状态就是以i为起点偷窃的最大价值
在某一个屋子面前 盗贼只有两种选择:偷或者不偷
1.偷的话 价值就是[当前房子的价值] + [下两个房子开始盗窃的最大价值]
不偷的话 价值就是下一个房子开始盗窃的最大价值

在这两个值中 选择最大记录在dp[i]中 就得到了以i为起点所能偷窃的最大价值
动态规划的起手式 找基础状态 在这题中 以终点为起点的最大价值一定是最好找的 因为终点不可能再往后偷窃了 所以设n为房子的总数量 dp[n-1]就是nums[n-1]
小偷只能盗窃这个房子 而不能跳过去选择下一个不存在的房子
那么就找到了动态规划的状态转移方程

盗窃当前的房子
robNow = nums[i] + dp[i+2] //当前房子的价值 + i+ 2下标房子为起点的最大价值
不抢当前房子 抢下一个房子
robNext = dp[i +1]// i + 1下标房子为起点的最大价值
// 两者选择最大值
dp[i] = Math.max(robNow,robNext),
// 并且从后往前求解

// [1,2,3,1]
function (nums){
    if(!nums.length){
        return 0;
    }
   let dp = []
   for(let i = nums.length -1; i <=0; i--){
        let robNow = nums[i]+([dp[i+2]]||0)  //1,3,2, 4
        let robNext = dp[i + 1]||0 
        dp[i] = Math.max(robNext,robNow) //1  3 3,4
   }
   return dp[0]
}

最后返回以0为起点开始打劫的最大价值即可

打家劫舍2

在上次打劫完一条街道之后和一圈房屋后,小偷又发现了一个新的可行窃的地区。这个地区只有一个入口,我们称之为“根”。 除了“根”之外,每栋房子有且只有一个“父“房子与之相连。一番侦察之后,聪明的小偷意识到“这个地方的所有房屋的排列类似于一棵二叉树”。 如果两个直接相连的房子在同一天晚上被打劫,房屋将自动报警。

计算在不触动警报的情况下,小偷一晚能够盗取的最高金额。

示例 1:

输入: [3,2,3,null,3,null,1]

 3
/ \

2 3
\
3 1
输出: 7
解释: 小偷一晚能够盗取的最高金额 = 3 + 3 + 1 = 7.
示例 2:

输入: [3,4,5,1,3,null,1]

 3
/ \

4 5
/ \
1 3 1
输出: 9
解释: 小偷一晚能够盗取的最高金额 = 4 + 5 = 9.

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/house-robber-iii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路
这题的题目上存在误导,并不是简单的跳一级去找最大值就可以,注意考虑这种情况:

 2
/ \

1 3
/
n 4
这种情况下并不要跳级,而是第二层的 3 和第三层的 4 是去凑成打劫的最优解。

所以此题的解法是从顶层节点开始

抢劫当前的节点,那么儿子层就没法抢劫了,只能抢劫孙子层。
不抢劫当前节点,那么可以抢劫儿子层。
这两者对比求出的最大值,就是最优结果。

自顶向下记忆法
// 3
// / \
// 2   3
// \   \
//  3   1
let memo = new WeakMap()
let rob = function (root){
    if(!root){
        return 0
    }
    let memorized = memo.get(root)
    if(memorized){
        return memorized
    }
    let notRob = rob(root.left) + rob(root.right)
    let robNow = (root.val||0)+(root.left?rob(root.left.left)+rob(root.left.right):0)+
    (root.right?rob(root.right.left)+rob(rob(root.right.right)):0)

    let max = Math.max(notRob,robNow)
    memo.set(root,max)
    return max
}

// 自底向上动态规划
// 上面的解法是自顶向下的,那么动态规划的自底向上解法应该怎么做呢?我们上一层的打劫最优解是依赖下一层的,所以显然我们应该先从最下层的求解。思考提取关键字「层序」、「自底向上」。

// 灵机一动,用递归回溯法配合BFS:

// 递归版的 BFS 先求出当前队列里所有的子节点,放入一个新的队列 subs 中,然后进一步 BFS 这个子节点队列 subs。

// 那么这个递归 subs 之后的一行,就代表递归后回溯的时机,我们把「动态规划」求解的部分放在递归函数的后面, 当 BFS 到达了最后一层后,发现没有节点可以继续 BFS 了,这个时候最底层的函数调用慢慢弹出栈,从最底层慢慢往上回溯,

// 那么 「动态规划」求解的部分就是「自底向上」的了,我们在上层中求最优解的时候,一定能取到下面层的最优解。

/**
 * @param {TreeNode} root
 * @return {number}
 */
let rob = function(root){
    if(!root) return 0
    let dp = new Map()
    dp.set(null,0)
    let dfs = (nodes)=>{
        if(!nodes.length){
            return
        }
        let subs = []
        for(let node of nodes){
            if(node.left){
                subs.push(node.left)
            }
            if(node.right){
                subs.push(node.right)
            }
        }
        bfs(subs)
        // 到达最底层后 最底层先开始dp
        // 再一层层回溯
        for(let node of nodes){
            // 打劫这个节点
            let robNow = node.val
            if(node.left){
                robNow += dp.get(node.left.left)
                robNow += dp.get(node.left.right)
            }
            if(node.right){
                robNow += dp.get(node.right.left)
                robNow += dp.get(node.right.right)
            }
            // 不打劫这个节点  打劫下一层
            let robNext = dp.get(node.left)+dp.get(node.right)
            dp.set(node,Math.max(robNow,robNext))
        }
    }
    bfs([root])
    return dp.get(root)
}

// 3
// /
// 4 5
// / \
// 1 3 1

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值