LeetCode算法心得——打家劫舍(记忆化搜索)

大家好,我是晴天学长,准备开始深入动态规划啦,先从记忆化搜索开始,需要的小伙伴可以关注支持一下哦!后续会继续更新的。


1) .打家劫舍


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

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

示例 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)。

提示:

1 <= nums.length <= 100
0 <= nums[i] <= 400


2) .算法思路

  • 纯dfs也可以,但是会超时。
  • 所以用到了记忆化搜索。
  • 用一个数组简单的记录搜索的答案进行返回。
  • 只关注现在的状态和上一步的状态。

3) .算法步骤

1.定义一个私有的数组 nums 和 memo,用于存储输入的房屋金额和记忆化搜索的结果。
2.创建 rob 方法来启动算法。在该方法中,初始化 nums 和 memo 数组,并将 memo 数组的所有元素初始化为 -1。
3.调用 dfs 方法,将起始索引设为 nums.length - 1(最后一个房屋),并返回结果。
4.在 dfs 方法中,首先检查是否已经搜索过当前索引 i 的结果。如果在 memo 数组中存在已计算的值,则直接返回该值作为结果,避免重复计算。
1)如果当前索引 i 小于 0,表示没有可选的房屋可偷窃,返回 0。
否则,根据动态规划的思想,选择在当前房屋偷窃或者不偷窃的最大值。2)使用递归调用 dfs 方法,分别计算不偷窃当前房屋的结果(即 dfs(i - 1))和偷窃当前房屋的结果(即 dfs(i - 2) + nums[i])。
3)取两种情况的最大值作为当前索引 i 的结果,并将其存储在 memo 数组中,以备后续使用。
4)返回当前索引 i 的结果作为最终答案。


4).代码示例

  class Solution {
        private int[] nums, memo;

        public int rob(int[] nums) {
            this.nums = nums;
            this.memo = new int[nums.length];
            Arrays.fill(memo, -1);
            return dfs(nums.length - 1);
        }

        private int dfs(int i) {
            //出口
            if (i < 0) return 0;
            // 看是否已经搜索过
            if (memo[i] != -1) {
                return memo[i];
            }
            int result = Math.max(dfs(i - 1), dfs(i - 2) + nums[i]);
            memo[i] = result;
            return result;
        }
    }

5).总结

  • 转移方程怎么写。

试题链接:

你可能感兴趣的:(算法,算法,leetcode,职场和发展)