打家劫舍问题(典型动态规划)

写在前

本类型题目要求不能偷盗相邻房屋,即对于当前房屋我们选择偷或者不偷两种状态,对应我们的最大收益值的更新(动态规划)。

1.打家劫舍(198-中)

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

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

示例

输入:[1,2,3,1]
输出:4
解释:偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。
     偷窃到的最高金额 = 1 + 3 = 4 。
     
输入:[2,7,9,3,1]
输出:12
解释:偷窃 1 号房屋 (金额 = 2), 偷窃 3 号房屋 (金额 = 9),接着偷窃 5 号房屋 (金额 = 1)。
     偷窃到的最高金额 = 2 + 9 + 1 = 12 。

思路:动态规划实现:由于相邻的两家店不能偷。对于第 n 家店,我们只能选择偷或者不偷,然后取两种情况的最大值。

  • dp[n] 数组:前 n 天能够带来的最大收益
  • 状态转移方程:dp[n] = Math.max(dp[n - 2] + nums[n - 1], dp[n - 1] )

代码:

public int rob1(int[] nums) {
    int n = nums.length;
    if (nums == null || n == 0) {
        return 0;
    }
    int[] dp = new int[n + 1];
    dp[0] = 0;
    dp[1] = nums[0];
    for (int i = 2; i < n + 1; i++) {
        dp[i] = Math.max(dp[i - 1], dp[i - 2] + nums[i - 1]);
    }
    return dp[n];
}

我们更新dp[i]的时候,只需要dp[i - 1]以及dp[i - 2]的信息(这里分别用pre2和pre1表示),再之前的信息就不需要了,所以可以进行空间优化。

public int rob2(int[] nums) {
    if (nums == null || nums.length == 0) {
        return 0;
    }
    int pre1 = 0, pre2 = 0;   
    for (int i : nums) {
        int tmp = Math.max(pre2, pre1 + i);
        pre1 = pre2;
        pre2 = tmp;
    }
    return pre2;
}

2.打家劫舍II(213-中)

题目描述:你是一个专业的小偷,计划偷窃沿街的房屋,每间房内都藏有一定的现金。这个地方所有的房屋都 围成一圈 ,这意味着第一个房屋和最后一个房屋是紧挨着的。同时,相邻的房屋装有相互连通的防盗系统,如果两间相邻的房屋在同一晚上被小偷闯入,系统会自动报警 。

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

示例

输入:[1,2,3,1]
输出:4
解释:偷窃 1 号房屋 (金额 = 1) ,然后偷窃 3 号房屋 (金额 = 3)。
     偷窃到的最高金额 = 1 + 3 = 4 。
     
输入:[2,7,9,3,1]
输出:12
解释:偷窃 1 号房屋 (金额 = 2), 偷窃 3 号房屋 (金额 = 9),接着偷窃 5 号房屋 (金额 = 1)。
     偷窃到的最高金额 = 2 + 9 + 1 = 12 。

思路:这道题区别案例1就是数组是环形,即偷了第一家就不能偷最后一家。

  • 偷第一家,也就是求出在前n - 1家中偷的最大收益,也就是不考虑最后一家的最大收益。

  • 不偷第一家,也就是求第2家到最后一家中偷的最大收益,也就是不考虑第一家的最大收益。

然后只需要返回上边两个最大收益中的衔接的即可。图示的话就是下边的两个范围。

X X X X X X
^       ^

X X X X X X
  ^       ^

dp数组和状态转移方程均相同,区别在于我们进行状态转移的区间有两个,取最大值。具体见代码。

代码:

public int rob(int[] nums) {
    //边界条件
    int n = nums.length;
    if (n == 0) return 0;
    if (n == 1) return nums[0];
    return Math.max(robHelper(nums, 0, n - 2), robHelper(nums, 1, n - 1));
}

private int robHelper(int[] nums, int start, int end) {
    int pre1 = 0, pre2 = 0;
    for (int i = start; i <= end; ++i) {
        int cur = Math.max(pre2, pre1 + nums[i]);
        pre1 = pre2;
        pre2 = cur;
    }
    return pre2;
}

3.打家劫舍III(337-中)

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

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

示例 :

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

     3
    / \
   2   3
    \   \ 
     3   1

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

思路:标准的动态规划,为了避免过多重复计算,用HashMap存放某一节点能偷到的最大值

法1:相互连接的父子节点不能同时偷,即转化为偷还是不偷root节点。

  • 偷root节点,可以偷root的孙子结点,只要节点不为空,以此类推...

  • 不偷root节点,可以偷root的左右子结点,只要左右节点不为空,以此类推...

法2:另外比较好的思路,以每个节点状态为基本状态,返回一个节点偷和不偷的最大值

  • 分别得到出当前节点,左右子节点偷与不偷的利润列表

  • 根据规则进行更新,res[0]代表偷,res[1]代表不偷,本质和法1类似。

代码1:比较常规思路。

public int rob(TreeNode root) {
    // 存放某一节点偷到的最大值
    HashMap<TreeNode, Integer> map = new HashMap<>();
    return rob1(root, map);
}

private int rob1(TreeNode root, HashMap<TreeNode, Integer> map) {
    if (root == null) return 0;
    if (map.containsKey(root)) return map.get(root);  
    
    // 计算偷root节点和孙子节点的情况
    int num = root.val;
    if (root.left != null) {
        num += rob1(root.left.left, map) + rob1(root.left.right, map);
    }
    if (root.right != null) {
        num += rob1(root.right.left, map) + rob1(root.right.right, map);
    }
    
    // 与只偷左右子节点比较取较大值
    int res = Math.max(num, rob1(root.left, map) + rob1(root.right, map));
    map.put(root, res);    
    return res;
}

代码2:每个节点都有两个状态,偷或者不偷,自底向上的递归求解。

public int rob(TreeNode root) {
    int[] result = rob2(root);
    return Math.max(result[0], result[1]);
}

// 返回值为偷或者不偷的root的最大值
public int[] rob2(TreeNode root) {
    //{0, 0}
    if (root == null) return new int[2];  
    int[] res = new int[2];
    // 左右孩子节点偷或者不偷情况
    int[] left = rob2(root.left);   
    int[] right = rob2(root.right);
    // res[0]代表偷,res[1]代表不偷
    res[0] = Math.max(left[0], left[1]) + Math.max(right[0], right[1]);
    res[1] = right[0] + left[0] + root.val;

    return res;
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 215,723评论 6 498
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,003评论 3 391
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 161,512评论 0 351
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,825评论 1 290
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,874评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,841评论 1 295
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,812评论 3 416
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,582评论 0 271
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,033评论 1 308
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,309评论 2 331
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,450评论 1 345
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,158评论 5 341
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,789评论 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,409评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,609评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,440评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,357评论 2 352

推荐阅读更多精彩内容