Struct seg_tree::segment_tree::LazySegmentTree

pub struct LazySegmentTree<T: LazyNode> { /* private fields */ }

Lazy segment tree with range queries and range updates. It uses O(n) space, assuming that each node uses O(1) space.

Implementations

impl<T: LazyNode + Clone> LazySegmentTree<T>

pub fn build(values: &[T]) -> Self

Builds lazy 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.

pub fn update(&mut self, i: usize, j: usize, value: <T as Node>::Value)

Updates the range [i,j] with value. It will panic if i or j is not in [0,n]. It has time complexity of O(log(n)), assuming that combine, update_lazy_value and update_lazy_value have constant time complexity.

pub fn query(&mut 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, update_lazy_value and update_lazy_value have constant time complexity.

pub fn lower_bound<F>(
    &self,
    predicate: F,
    g: fn(_: &<T as Node>::Value, _: <T as Node>::Value) -> <T as Node>::Value,
    value: <T as Node>::Value
) -> usize where
    F: Fn(&<T as Node>::Value, &<T as Node>::Value) -> bool, 

A method that finds the smallest prefix¹ u such that predicate(u.value(), value) is true. The following must be true:

• predicate is monotonic over prefixes².
• g will satisfy the following, given segments [i,j] and [i,k] with j<k we have that predicate([i,k].value(),value) implies predicate([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 = LazySegmentTree::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 = LazySegmentTree::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

---

1. A prefix is a segment of the form [0,i]. ↩

2. Given two prefixes u and v if u is contained in v then predicate(u.value(), value) implies predicate(v.value(), value). ↩