update from atlas

02-array-seq/bisect_demo.py

@@ -0,0 +1,27 @@
+# BEGIN BISECT_DEMO
+import bisect
+import sys
+
+HAYSTACK = [1, 4, 5, 6, 8, 12, 15, 20, 21, 23, 23, 26, 29, 30]
+NEEDLES = [0, 1, 2, 5, 8, 10, 22, 23, 29, 30, 31]
+
+ROW_FMT = '{0:2d} @ {1:2d}    {2}{0:<2d}'
+
+def demo(bisect_fn):
+    for needle in reversed(NEEDLES):
+        position = bisect_fn(HAYSTACK, needle)  # <1>
+        offset = position * '  |'  # <2>
+        print(ROW_FMT.format(needle, position, offset))  # <3>
+
+if __name__ == '__main__':
+
+    if sys.argv[-1] == 'left':    # <4>
+        bisect_fn = bisect.bisect_left
+    else:
+        bisect_fn = bisect.bisect
+
+    print('DEMO:', bisect_fn.__name__)  # <5>
+    print('haystack ->', ' '.join('%2d' % n for n in HAYSTACK))
+    demo(bisect_fn)
+
+# END BISECT_DEMO