From 908fd6c8e0ea85d682d8a7d77adceffddefaa4bf Mon Sep 17 00:00:00 2001
From: CyC2018 <zhengyc101@163.com>
Date: Mon, 19 Apr 2021 00:32:56 +0800
Subject: [PATCH] auto commit

---
 notes/Leetcode 题解 - 树.md | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/notes/Leetcode 题解 - 树.md b/notes/Leetcode 题解 - 树.md
index 410553f7..c0243939 100644
--- a/notes/Leetcode 题解 - 树.md	
+++ b/notes/Leetcode 题解 - 树.md	
@@ -427,13 +427,18 @@ Maximum amount of money the thief can rob = 3 + 3 + 1 = 7.
 ```
 
 ```java
+Map<TreeNode, Integer> cache = new HashMap<>(); 
+
 public int rob(TreeNode root) {
     if (root == null) return 0;
+    if (cache.containsKey(root)) return cache.get(root);
     int val1 = root.val;
     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);
     int val2 = rob(root.left) + rob(root.right);
-    return Math.max(val1, val2);
+    int res = Math.max(val1, val2);
+    cache.put(root, res);
+    return res;
 }
 ```