Skip to content

Commit 8e16739

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

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
@@ -83,6 +83,7 @@
8383
Query both the children and add their results and return that
8484
"""
8585

86+
8687
class SegmentTree:
8788
def __init__(self, arr, merge_func, default):
8889
"""
@@ -97,7 +98,7 @@ def __init__(self, arr, merge_func, default):
9798
self.n = len(arr)
9899

99100
# while self.n is not a power of two
100-
while (self.n & (self.n-1)) != 0:
101+
while (self.n & (self.n - 1)) != 0:
101102
self.n += 1
102103
self.arr.append(default)
103104

@@ -109,8 +110,9 @@ def __init__(self, arr, merge_func, default):
109110
self.segment_tree[self.n + i] = arr[i]
110111

111112
for i in range(self.n - 1, 0, -1):
112-
self.segment_tree[i] = self.merge_func(self.segment_tree[2 * i],
113-
self.segment_tree[2 * i + 1])
113+
self.segment_tree[i] = self.merge_func(
114+
self.segment_tree[2 * i], self.segment_tree[2 * i + 1]
115+
)
114116

115117
def update(self, index, value):
116118
"""
@@ -120,8 +122,9 @@ def update(self, index, value):
120122

121123
while index >= 1:
122124
index //= 2 # Go to the parent of index
123-
self.segment_tree[index] = self.merge_func(self.segment_tree[2 * index],
124-
self.segment_tree[2 * index + 1])
125+
self.segment_tree[index] = self.merge_func(
126+
self.segment_tree[2 * index], self.segment_tree[2 * index + 1]
127+
)
125128

126129
def query(self, left, right, node_index=1, node_left=0, node_right=None):
127130
"""
@@ -148,5 +151,5 @@ def query(self, left, right, node_index=1, node_left=0, node_right=None):
148151
# of the query values of both the children nodes
149152
return self.merge_func(
150153
self.query(left, right, node_index * 2, node_left, mid),
151-
self.query(left, right, node_index * 2 + 1, mid + 1, node_right)
154+
self.query(left, right, node_index * 2 + 1, mid + 1, node_right),
152155
)

0 commit comments

Comments
 (0)