feat: created mo's algolithm library

Author: hidehic0

code/main.py

@@ -1304,6 +1304,57 @@ def lcp_sum(self, value: str) -> int:
         return result
 
 
+import math
+from typing import Any, Callable, List
+
+
+def mo_algorithm(
+    N: int,
+    queries: List[Any],
+    add: Callable[[int], Any],
+    delete: Callable[[int], Any],
+    getvalue: Callable[[], Any],
+) -> List[Any]:
+    """
+    Mo's algorithmの関数
+    queriesは、(左端, 右端)で1-indexed
+    addはあるindexが追加される時の値を現在の値にする
+    deleteはあるindexが削除される時の値を現在の値にする
+    getvalueは現在の値を返す
+    """
+    Q = len(queries)
+    res = [None] * Q
+    M = int(max(1, 1.0 * N / max(1, math.sqrt(Q * 2.0 / 3.0))))
+
+    queries = [(l, r, i) for i, (l, r) in enumerate(queries)]
+    queries.sort(key=lambda x: (x[0] // M, x[1] if (x[0] // M) % 2 == 0 else -x[1]))
+
+    cl, cr = 0, -1
+
+    for l, r, ind in queries:
+        l -= 1
+        r -= 1
+        while cl > l:
+            cl -= 1
+            add(cl)
+
+        while cr < r:
+            cr += 1
+            add(cr)
+
+        while cl < l:
+            delete(cl)
+            cl += 1
+
+        while cr > r:
+            delete(cr)
+            cr -= 1
+
+        res[ind] = getvalue()
+
+    return res
+
+
 from typing import List
 
 

libs/mo.py

@@ -0,0 +1,49 @@
+import math
+from typing import Any, Callable, List
+
+
+def mo_algorithm(
+    N: int,
+    queries: List[Any],
+    add: Callable[[int], Any],
+    delete: Callable[[int], Any],
+    getvalue: Callable[[], Any],
+) -> List[Any]:
+    """
+    Mo's algorithmの関数
+    queriesは、(左端, 右端)で1-indexed
+    addはあるindexが追加される時の値を現在の値にする
+    deleteはあるindexが削除される時の値を現在の値にする
+    getvalueは現在の値を返す
+    """
+    Q = len(queries)
+    res = [None] * Q
+    M = int(max(1, 1.0 * N / max(1, math.sqrt(Q * 2.0 / 3.0))))
+
+    queries = [(l, r, i) for i, (l, r) in enumerate(queries)]
+    queries.sort(key=lambda x: (x[0] // M, x[1] if (x[0] // M) % 2 == 0 else -x[1]))
+
+    cl, cr = 0, -1
+
+    for l, r, ind in queries:
+        l -= 1
+        r -= 1
+        while cl > l:
+            cl -= 1
+            add(cl)
+
+        while cr < r:
+            cr += 1
+            add(cr)
+
+        while cl < l:
+            delete(cl)
+            cl += 1
+
+        while cr > r:
+            delete(cr)
+            cr -= 1
+
+        res[ind] = getvalue()
+
+    return res

merge_file.bash

@@ -7,7 +7,7 @@ echo "新しいmain.py作成完了"
 # テンプレ
 # /bin/cat python/<filename>.py >> code/main.py
 
-for file_name in "import.py" "standard_input.py" "math_func.py" "array_create.py" "binary_search.py" "modint.py" "yn_func.py" "grid.py" "coordinates_to_id.py" "dijkstra.py" "get_path.py" "dp.py" "coordinate_compression.py" "memo.py" "lca_weight.py" "rollinghash.py" "graph.py" "unionfind.py" "potential_unionfind.py" "heap.py" "trie.py" "bit.py" "dis_lib.py" "alias.py" "utils.py"; do
+for file_name in "import.py" "standard_input.py" "math_func.py" "array_create.py" "binary_search.py" "modint.py" "yn_func.py" "grid.py" "coordinates_to_id.py" "dijkstra.py" "get_path.py" "dp.py" "coordinate_compression.py" "memo.py" "lca_weight.py" "rollinghash.py" "graph.py" "unionfind.py" "potential_unionfind.py" "heap.py" "trie.py" "mo.py" "bit.py" "dis_lib.py" "alias.py" "utils.py"; do
 
   cat libs/${file_name} >>code/main.py
 done

mise.toml

@@ -10,3 +10,4 @@ run = "./pre_commit.bash"
 
 [env]
 _.python.venv = { path = ".venv" }
+PYTHON_PATH = "."

pyproject.toml

@@ -11,14 +11,3 @@ line-length = 88
 
 [tool.ruff.lint]
 select = ["C9", "E", "F", "W", "I", "ANN"]
-ignore = [
-  "F401",
-  "E402",
-  "E741",
-  "F405",
-  "F403",
-  "I001",
-  "E722",
-  "E501",
-  "F811",
-]