Skip to content

Commit 4484382

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent f64233a commit 4484382

File tree

1 file changed

+16
-5
lines changed

1 file changed

+16
-5
lines changed

data_structures/persistent_segment_tree.py

+16-5
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,13 @@
11
from __future__ import annotations
22

3+
34
class Node:
45
def __init__(self, value: int = 0) -> None:
56
self.value: int = value
67
self.left: Node | None = None
78
self.right: Node | None = None
89

10+
911
class PersistentSegmentTree:
1012
def __init__(self, arr: list[int]) -> None:
1113
self.n: int = len(arr)
@@ -52,7 +54,9 @@ def update(self, version: int, index: int, value: int) -> int:
5254
self.roots.append(new_root)
5355
return len(self.roots) - 1
5456

55-
def _update(self, node: Node | None, start: int, end: int, index: int, value: int) -> Node:
57+
def _update(
58+
self, node: Node | None, start: int, end: int, index: int, value: int
59+
) -> Node:
5660
if node is None:
5761
raise ValueError("Cannot update a None node")
5862

@@ -69,7 +73,9 @@ def _update(self, node: Node | None, start: int, end: int, index: int, value: in
6973
new_node.left = node.left
7074
new_node.right = self._update(node.right, mid + 1, end, index, value)
7175

72-
new_node.value = new_node.left.value + (new_node.right.value if new_node.right else 0)
76+
new_node.value = new_node.left.value + (
77+
new_node.right.value if new_node.right else 0
78+
)
7379

7480
return new_node
7581

@@ -90,7 +96,9 @@ def query(self, version: int, left: int, right: int) -> int:
9096
"""
9197
return self._query(self.roots[version], 0, self.n - 1, left, right)
9298

93-
def _query(self, node: Node | None, start: int, end: int, left: int, right: int) -> int:
99+
def _query(
100+
self, node: Node | None, start: int, end: int, left: int, right: int
101+
) -> int:
94102
if node is None:
95103
return 0
96104

@@ -99,12 +107,15 @@ def _query(self, node: Node | None, start: int, end: int, left: int, right: int)
99107
if left <= start and right >= end:
100108
return node.value
101109
mid = (start + end) // 2
102-
return (self._query(node.left, start, mid, left, right) +
103-
self._query(node.right, mid + 1, end, left, right))
110+
return self._query(node.left, start, mid, left, right) + self._query(
111+
node.right, mid + 1, end, left, right
112+
)
113+
104114

105115
# Running the doctests
106116
if __name__ == "__main__":
107117
import doctest
118+
108119
print("Running doctests...")
109120
result = doctest.testmod()
110121
print(f"Ran {result.attempted} tests, {result.failed} failed.")

0 commit comments

Comments
 (0)