【牛客 NC176】打家劫舍(一)

最新推荐文章于 2022-12-09 10:37:01 发布

hjhcos

最新推荐文章于 2022-12-09 10:37:01 发布

阅读量505

点赞数

分类专栏：力扣文章标签： java 算法动态规划

hjhcos

本文链接：https://blog.csdn.net/hjh_cos/article/details/123246908

版权

力扣专栏收录该内容

5 篇文章 0 订阅

订阅专栏

文章目录

📚问题描述
- ❌错误解题思路
- ✔️动态规划解法
📚动态规划
参考资料

📚问题描述

原问题地址：打家劫舍(一)_牛客网 (nowcoder.com)

知识点：动态规划

难度：⭐⭐⭐

要求：

时间限制：2秒
空间限制：256M

描述

你是一个经验丰富的小偷，准备偷沿街的一排房间，每个房间都存有一定的现金，为了防止被发现，你不能偷相邻的两家，即，如果偷了第一家，就不能再偷第二家；如果偷了第二家，那么就不能偷第一家和第三家。

给定一个整数数组 nums，数组中的元素表示每个房间存有的现金数额，请你计算在不被发现的前提下最多的偷窃金额。

数据范围：数组长度满足 $ 1 ≤ n ≤ 2\times 10^5 $ ，数组中每个值满足 $ 1 ≤ num[i] ≤ 5000 $。

❌错误解题思路

当我看到这个题目的时候，考虑的非常简单，无非就是通过检查两家之间的金额差距，来决定最后的金额。

public class Solution {
    public int rob (int[] nums) {
        // 偶数就从第二个开始
        // 奇数就从第一个开始
        // 后来我看见第三个示例
        // 打破了我这种想法 终究是我太年轻了
        if(nums.length==1)return nums[0];
        // 这里通过相邻两家钱数差距来决定最后能偷取的最大钱数
        int num = 0;    // 差数
        long res = 0l;    // 总数
        int count = 1;  // 索引
        while(count<nums.length){
            res += nums[count-1];
            num += nums[count-1]-nums[count];
            count +=2;
        }
        if(nums.length%2==1){
            num += nums[count-1];
            res += nums[count-1];
        }
        res = num>0?res:res+Math.abs(num);
        return (int)res;
    }
}

10/14 组用例通过运行时间265ms 占用内存30388KB 预期输出 293318815 实际输出 250223388

没错上面是无法通过所有示例的，因为没有考虑有某家的钱数高于相邻几家钱数的总和这种情况。

例如：[2, 3, 10, 5, 2, 9] 这种情况，那么你觉得如何才能让小偷偷到最多的钱数❓

按上面那种方式比较两家的钱数差距来决定最后偷取的总钱数 17 = 3 + 5 + 9，显然是这个答案是不正确的，实际上小偷可以偷取 21 = 2 + 10 + 9。

✔️动态规划解法

其实一开始我并没有看到知识点是动态规划，就直接去做了。

这也导致程序出现问题，不能通过测试。

public class Solution {
    public int rob (int[] nums) {
        int[][] dp = new int[2][2];
        dp[0][0] = 0;
        dp[0][1] = nums[0];
        for(int i = 1; i<nums.length;i++){
            // 当前这家小偷不偷 取上次偷或不偷中的金额最大的那个
            dp[1][0] = dp[0][0] > dp[0][1]?dp[0][0]:dp[0][1];
            // 当前这家小偷偷 只能取上次不偷的金额
            dp[1][1] = dp[0][0] + nums[i];
            // 我这里并没有保留小偷每经过一家偷与不偷的情况
            // 将当前偷与不偷定义为上一次偷与不偷的金额 为小偷到下一家偷与不偷做准备
            dp[0][0] = dp[1][0];
            dp[0][1] = dp[1][1];
        }
        return dp[0][0] > dp[0][1]?dp[0][0]:dp[0][1];
    }
}