pub struct Recursive<T> { /* private fields */ }
Expand description
Segment tree with range queries and point updates.
It uses O(n)
space, assuming that each node uses O(1)
space.
Note if you don’t need to use lower_bound
, just use the SegmentTree
it uses half the memory and it’s more performant.
Implementations§
Source§impl<T> Recursive<T>
impl<T> Recursive<T>
Sourcepub fn build(values: &[T]) -> Self
pub fn build(values: &[T]) -> Self
Builds segment tree from slice, each element of the slice will correspond to a leaf of the segment tree.
It has time complexity of O(n*log(n))
, assuming that combine has constant time complexity.
Sourcepub fn update(&mut self, p: usize, value: &<T as Node>::Value)
pub fn update(&mut self, p: usize, value: &<T as Node>::Value)
Sets the p-th element of the segment tree to value T and update the segment tree correspondingly.
It will panic if p is not in [0,n)
It has time complexity of O(log(n))
, assuming that combine has constant time complexity.
Sourcepub fn query(&self, left: usize, right: usize) -> Option<T>
pub fn query(&self, left: usize, right: usize) -> Option<T>
Returns the result from the range [left,right]
.
It returns None if and only if range is empty.
It will panic if left
or right
are not in [0,n).
It has time complexity of O(log(n))
, assuming that combine has constant time complexity.
Sourcepub fn lower_bound<F, G>(
&self,
predicate: F,
g: G,
value: <T as Node>::Value,
) -> usize
pub fn lower_bound<F, G>( &self, predicate: F, g: G, value: <T as Node>::Value, ) -> usize
A method that finds the smallest prefix1 u
such that predicate(u.value(), value)
is true
. The following must be true:
predicate
is monotonic over prefixes2.g
will satisfy the following, given segments[i,j]
and[i,k]
withj<k
we have thatpredicate([i,k].value(),value)
impliespredicate([j+1,k].value(),g([i,j].value(),value))
.
These are two examples, the first is finding the smallest prefix which sums at least some value.
let predicate = |left_value: &usize, value: &usize|{*left_value >= *value}; // Is the sum greater or equal to value?
let g = |left_node: &usize, value: usize|{value - *left_node}; // Subtract the sum of the prefix.
let seg_tree = Recursive::build(&nodes); // [0,1,2,3,4,5,6,7,8,9] with Sum<usize> nodes
let index = seg_tree.lower_bound(predicate, g, 3); // Will return 2 as sum([0,1,2])>=3
The second is finding the position of the smallest value greater or equal to some value.
let predicate = |left_value:&usize, value:&usize|{*left_value>=*value}; // Is the maximum greater or equal to value?
let g = |_left_node:&usize,value:usize|{value}; // Do nothing
let seg_tree = Recursive::build(&nodes); // [0,1,2,3,4,5,6,7,8,9] with Max<usize> nodes
let index = seg_tree.lower_bound(predicate, g, 3); // Will return 3 as 3>=3