maximum-depth-of-binary-tree

Author: changhyumm

maximum-depth-of-binary-tree/changhyumm.py

@@ -0,0 +1,20 @@
+# Definition for a binary tree node.
+# class TreeNode:
+#     def __init__(self, val=0, left=None, right=None):
+#         self.val = val
+#         self.left = left
+#         self.right = right
+class Solution:
+    def maxDepth(self, root: Optional[TreeNode]) -> int:
+        if root is None:
+            return 0
+        stack = [[root, 1]]
+        max_depth = 1
+        while stack:
+            node, cur_depth = stack.pop()
+            max_depth = max(max_depth, cur_depth)
+            if node.left:
+                stack.append([node.left, cur_depth + 1])
+            if node.right:
+                stack.append([node.right, cur_depth + 1])
+        return max_depth