StarMirror/CS-Notes
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix issue 1074

Browse Source
This commit is contained in:
hu.fwh 2021-04-14 02:42:10 +08:00
parent 156f7a67f4
commit 24c6cccbc0
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
notes/Leetcode 题解 - 树.md
View File

@ -428,12 +428,17 @@ Maximum amount of money the thief can rob = 3 + 3 + 1 = 7.
```java ```java
public int rob(TreeNode root) { public int rob(TreeNode root) {
if (root == null) return 0; int[] num = dfs(root);
int val1 = root.val; return Math.max(num[0], num[1]);
if (root.left != null) val1 += rob(root.left.left) + rob(root.left.right); }
if (root.right != null) val1 += rob(root.right.left) + rob(root.right.right); private int[] dfs(TreeNode node) {
int val2 = rob(root.left) + rob(root.right); if (node == null) return new int[2];
return Math.max(val1, val2); int[] left = dfs(node.left);
int[] right = dfs(node.right);
int[] res = new int[2];
res[0] = left[1] + right[1] + node.val;
res[1] = Math.max(left[0], left[1]) + Math.max(right[0], right[1]);
return res;
} }
``` ```