Create 0337-house-robber-iii.java

Arrrsh · May 31, 2023 · 2cdea0c · 2cdea0c
1 parent f02c591
commit 2cdea0c
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/java/0337-house-robber-iii.java b/java/0337-house-robber-iii.java
@@ -0,0 +1,19 @@
+class Solution {
+ public int[] dfs(TreeNode root){
+ if(root == null) return new int[2];
+
+ int []left = dfs(root.left);
+ int []right = dfs(root.right);
+
+ int []res = new int[2];
+
+ res[0] = left[1] + right[1] + root.val; //with Root
+ res[1] = Math.max(left[0], left[1]) + Math.max(right[0], right[1]); //without Root
+
+ return res;
+ }
+ public int rob(TreeNode root) {
+ int []ans = dfs(root);
+ return Math.max(ans[0], ans[1]);
+ }
+}