Bootstrap todo-next

2 files changed, 35 insertions, 0 deletions

diff --git a/src/util/array.py b/src/util/array.py
new file mode 100644
index 0000000..7dd0357
--- /dev/null
+++ b/src/util/array.py
@@ -0,0 +1,5 @@
+def insert_position(x, xs, is_reversed: bool) -> int:
+    for i, y in enumerate(xs):
+        if is_reversed and x > y or not is_reversed and x < y:
+            return i
+    return len(xs)
diff --git a/src/util/range.py b/src/util/range.py
new file mode 100644
index 0000000..c75232c
--- /dev/null
+++ b/src/util/range.py
@@ -0,0 +1,30 @@
+from typing import NamedTuple
+
+class Range(NamedTuple):
+    start: int
+    length: int
+
+def from_indexes(indexes):
+    ranges = []
+    curr_range_start = 0
+    curr_range_len = 0
+
+    last_index = -1
+
+    for index in sorted(indexes):
+        if index == curr_range_start + curr_range_len:
+            curr_range_len += 1
+        else:
+            if curr_range_len > 0:
+                ranges.append(Range(
+                    start = curr_range_start,
+                    length = curr_range_len))
+            curr_range_start = index
+            curr_range_len = 1
+
+    if curr_range_len > 0:
+        ranges.append(Range(
+            start = curr_range_start,
+            length = curr_range_len))
+
+    return ranges