Skip to content

Commit 4e95c90

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

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)
@@ -33,7 +35,9 @@ def update(self, version: int, index: int, value: int) -> int:
3335
self.roots.append(new_root)
3436
return len(self.roots) - 1
3537

36-
def _update(self, node: Node | None, start: int, end: int, index: int, value: int) -> Node:
38+
def _update(
39+
self, node: Node | None, start: int, end: int, index: int, value: int
40+
) -> Node:
3741
"""
3842
Updates the node for the specified index and value and returns the new node.
3943
"""
@@ -53,7 +57,9 @@ def _update(self, node: Node | None, start: int, end: int, index: int, value: in
5357
new_node.left = node.left # Ensure left node is the same as the original
5458
new_node.right = self._update(node.right, mid + 1, end, index, value)
5559

56-
new_node.value = new_node.left.value + (new_node.right.value if new_node.right else 0)
60+
new_node.value = new_node.left.value + (
61+
new_node.right.value if new_node.right else 0
62+
)
5763

5864
return new_node
5965

@@ -63,7 +69,9 @@ def query(self, version: int, left: int, right: int) -> int:
6369
"""
6470
return self._query(self.roots[version], 0, self.n - 1, left, right)
6571

66-
def _query(self, node: Node | None, start: int, end: int, left: int, right: int) -> int:
72+
def _query(
73+
self, node: Node | None, start: int, end: int, left: int, right: int
74+
) -> int:
6775
"""
6876
Queries the sum of values in the range [left, right] for the given node.
6977
"""
@@ -72,12 +80,15 @@ def _query(self, node: Node | None, start: int, end: int, left: int, right: int)
7280
if left <= start and right >= end:
7381
return node.value
7482
mid = (start + end) // 2
75-
return (self._query(node.left, start, mid, left, right) +
76-
self._query(node.right, mid + 1, end, left, right))
83+
return self._query(node.left, start, mid, left, right) + self._query(
84+
node.right, mid + 1, end, left, right
85+
)
86+
7787

7888
# Running the doctests
7989
if __name__ == "__main__":
8090
import doctest
91+
8192
print("Running doctests...")
8293
result = doctest.testmod()
8394
print(f"Ran {result.attempted} tests, {result.failed} failed.")

0 commit comments

Comments
 (0)