feat: 双対セグ木ライブラリを作成

Author: hidehic0

libs/dual_segtree.py

@@ -0,0 +1,48 @@
+from typing import Any, Callable
+
+
+class DualSegmentTree:
+    def __init__(self, op: Callable[[Any, Any], Any], e: Any, n: int) -> None:
+        """
+        区間作用/一点取得のセグメント木
+        opは区間作用用の関数
+        eは初期値
+        vは長さ
+        """
+        self._op: Callable[[Any, Any], Any] = op
+        self._e: Any = e
+        self._n: int = n
+        self.n: int = 1 << (n - 1).bit_length()
+        self.data = [e] * (self.n * 2)
+
+    def apply(self, l, r, x) -> None:
+        """
+        区間[l,r)にxを適用
+        """
+        l += self.n
+        r += self.n
+
+        while l < r:
+            if l & 1:
+                self.data[l] = self._op(self.data[l], x)
+                l += 1
+
+            if r & 1:
+                self.data[r - 1] = self._op(self.data[r - 1], x)
+
+            l >>= 1
+            r >>= 1
+
+    def get(self, p: int) -> Any:
+        """
+        pの値を取得する
+        """
+
+        res = self._e
+        p += self.n
+
+        while p:
+            res = self._op(res, self.data[p])
+            p >>= 1
+
+        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" "mo.py" "square-division.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" "square-division.py" "bit.py" "dual_segtree.py" "dis_lib.py" "alias.py" "utils.py"; do
 
   cat libs/${file_name} >>code/main.py
 done

mise.toml

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

pyproject.toml

@@ -5,11 +5,13 @@ description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
-    "debugpy>=1.8.14",
+  "debugpy>=1.8.14",
 ]
 
 [tool.ruff]
 line-length = 88
 
 [tool.ruff.lint]
 select = ["C9", "E", "F", "W", "I", "ANN"]
+ignore = ["E741", "ANN001"]
+

tests/ABC408C.py

@@ -0,0 +1,15 @@
+from libs.dual_segtree import DualSegmentTree
+
+N, M = map(int, input().split())
+seg = DualSegmentTree(lambda a, b: a + b, 0, N)
+
+for _ in [0] * M:
+    l, r = map(int, input().split())
+    seg.apply(l - 1, r, 1)
+
+ans = 1000000000000000000
+
+for i in range(N):
+    ans = min(ans, seg.get(i))
+
+print(ans)