LeetCode 刷题笔记 之 House Robber III

xiaoxiao2021-02-28  8

题目如下:

The thief has found himself a new place for his thievery again. There is only one entrance to this area, called the "root." Besides the root, each house has one and only one parent house. After a tour, the smart thief realized that "all houses in this place forms a binary tree". It will automatically contact the police if two directly-linked houses were broken into on the same night.

Determine the maximum amount of money the thief can rob tonight without alerting the police.

Example 1:

3 / \ 2 3 \ \ 3 1 Maximum amount of money the thief can rob =  3  +  3  +  1  =  7 .

Example 2:

3 / \ 4 5 / \ \ 1 3 1

Maximum amount of money the thief can rob = 4 + 5 = 9.

解答如下:

本来想简单地用递归去查找奇数层的和以及偶数层的和。但是忽略了。跨两层的和可能更大,比如:7----2----1----4。所以这题的正解还是动态规划。

class Solution { public int rob(TreeNode root) { return dfs(root)[1]; } private int[] dfs(TreeNode root) { int[] rob ={0, 0}; if(root != null) { int[] robLeft = dfs(root.left); int[] robRight = dfs(root.right); rob[0] = robLeft[1] + robRight[1]; //左子树的最大值,右子树的最大值 rob[1] = Math.max(robLeft[0] + robRight[0] + root.val, rob[0]); //要么是子树层,要么是本层+子树的子树层。 } return rob; } }
转载请注明原文地址: https://www.6miu.com/read-1750314.html

最新回复(0)