Merge pull request neetcode-gh#482 from MeViMo/patch-1

Ahmad-A0 · web-flow · commit 6aecc8c42d23 · 2022-08-08T13:54:57.000+01:00
Update 543-Diameter-of-Binary-Tree.py
diff --git a/python/543-Diameter-of-Binary-Tree.py b/python/543-Diameter-of-Binary-Tree.py
@@ -5,17 +5,19 @@
 #         self.left = left
 #         self.right = right
 class Solution:
-    def diameterOfBinaryTree(self, root: TreeNode) -> int:
-        res = [0]
+    def diameterOfBinaryTree(self, root: Optional[TreeNode]) -> int:
+        res = 0
 
         def dfs(root):
+            nonlocal res
+
             if not root:
-                return -1
+                return 0
             left = dfs(root.left)
             right = dfs(root.right)
-            res[0] = max(res[0], 2 + left + right)
+            res = max(res, left + right)
 
             return 1 + max(left, right)
 
         dfs(root)
-        return res[0]
+        return res
