-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathsegmenttree.py
47 lines (37 loc) · 1.15 KB
/
segmenttree.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
class SegmentTree:
def __init__(self, values):
self.data = [0 for _ in values] + values
self.n = len(values)
for idx in reversed(range(1, self.n)):
self.data[idx] = min(self.data[2*idx], self.data[2*idx+1])
def update(self, idx, value):
idx += self.n
self.data[idx] = value
while idx > 1:
idx //= 2
self.data[idx] = min(self.data[2*idx], self.data[2*idx+1])
def minimum(self, left, right):
left += self.n
right += self.n
minimum = self.data[left]
while left < right:
if left % 2:
minimum = min(minimum, self.data[left])
left += 1
if right % 2:
right -= 1
minimum = min(minimum, self.data[right])
left //= 2
right //= 2
return minimum
if __name__ == '__main__':
st = SegmentTree([1, 5, 3, 7, 5])
print(st.minimum(0, 3))
print(st.minimum(1, 3))
print(st.minimum(1, 2))
st.update(0,3)
st.update(2,4)
print()
print(st.minimum(0, 3))
print(st.minimum(1, 3))
print(st.minimum(1, 2))