Skip to content

Commit 280bc4f

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

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

searches/segment_tree.py

+9-6
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,7 @@
8181
Query both the children and add their results and return that
8282
"""
8383

84+
8485
class SegmentTree:
8586
def __init__(self, arr, merge_func, default):
8687
"""
@@ -95,7 +96,7 @@ def __init__(self, arr, merge_func, default):
9596
self.n = len(arr)
9697

9798
# while self.n is not a power of two
98-
while (self.n & (self.n-1)) != 0:
99+
while (self.n & (self.n - 1)) != 0:
99100
self.n += 1
100101
self.arr.append(default)
101102

@@ -107,8 +108,9 @@ def __init__(self, arr, merge_func, default):
107108
self.segment_tree[self.n + i] = arr[i]
108109

109110
for i in range(self.n - 1, 0, -1):
110-
self.segment_tree[i] = self.merge_func(self.segment_tree[2 * i],
111-
self.segment_tree[2 * i + 1])
111+
self.segment_tree[i] = self.merge_func(
112+
self.segment_tree[2 * i], self.segment_tree[2 * i + 1]
113+
)
112114

113115
def update(self, index, value):
114116
"""
@@ -118,8 +120,9 @@ def update(self, index, value):
118120

119121
while index >= 1:
120122
index //= 2 # Go to the parent of index
121-
self.segment_tree[index] = self.merge_func(self.segment_tree[2 * index],
122-
self.segment_tree[2 * index + 1])
123+
self.segment_tree[index] = self.merge_func(
124+
self.segment_tree[2 * index], self.segment_tree[2 * index + 1]
125+
)
123126

124127
def query(self, left, right, node_index=1, node_left=0, node_right=None):
125128
"""
@@ -146,5 +149,5 @@ def query(self, left, right, node_index=1, node_left=0, node_right=None):
146149
# of the query values of both the children nodes
147150
return self.merge_func(
148151
self.query(left, right, node_index * 2, node_left, mid),
149-
self.query(left, right, node_index * 2 + 1, mid + 1, node_right)
152+
self.query(left, right, node_index * 2 + 1, mid + 1, node_right),
150153
)

0 commit comments

Comments
 (0)