Create: 0103-binary-tree-level-order-traversal.py

sahasourav17 · sahasourav17 · commit bae51a22bf6e · 2023-08-24T21:09:46.000+06:00
diff --git a/python/0103-binary-tree-zigzag-level-order-traversal.py b/python/0103-binary-tree-zigzag-level-order-traversal.py
@@ -0,0 +1,24 @@
+# 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 zigzagLevelOrder(self, root: Optional[TreeNode]) -> List[List[int]]:
+        if root is None:
+            return
+        result, zigzagDirection = [], 1
+        q = [root]
+        while q:
+            level, queueLength = [], len(q)
+            for i in range(queueLength):
+                node = q.pop(0)
+                level.append(node.val)
+                if node.left:
+                    q.append(node.left)
+                if node.right:
+                    q.append(node.right)
+            result.append(level[::zigzagDirection])
+            zigzagDirection *= -1
+        return result
