Create path-in-zigzag-labelled-binary-tree.py

Author: kamyu104

Python/path-in-zigzag-labelled-binary-tree.py

@@ -0,0 +1,17 @@
+# Time:  O(logn)
+# Space: O(logn)
+
+class Solution(object):
+    def pathInZigZagTree(self, label):
+        """
+        :type label: int
+        :rtype: List[int]
+        """
+        count = 2**label.bit_length()
+        result = []
+        while label >= 1:
+            result.append(label)
+            label = ((count//2) + ((count-1)-label)) // 2
+            count //= 2
+        result.reverse()
+        return result