segtree.cpp File Reference

Implementation of [Segment Tree] (https://en.wikipedia.org/wiki/Segment_tree) data structure. More...

#include <cassert>
#include <cmath>
#include <cstdint>
#include <iostream>
#include <vector>

Include dependency graph for segtree.cpp:

Go to the source code of this file.

Functions
void	ConsTree (const std::vector< int64_t > &arr, std::vector< int64_t > *segtree, uint64_t low, uint64_t high, uint64_t pos)
	for std::vector
int64_t	query (std::vector< int64_t > *segtree, std::vector< int64_t > *lazy, uint64_t qlow, uint64_t qhigh, uint64_t low, uint64_t high, uint64_t pos)
	Returns the sum of all elements in a range.
void	update (std::vector< int64_t > *segtree, std::vector< int64_t > *lazy, int64_t start, int64_t end, int64_t delta, uint64_t low, uint64_t high, uint64_t pos)
	Updates a range of the segment tree.
static void	test ()
	Self-test implementation.
int	main ()
	Main function.

Detailed Description

Implementation of [Segment Tree] (https://en.wikipedia.org/wiki/Segment_tree) data structure.

A segment tree, also known as a statistic tree, is a tree data structure used for storing information about intervals, or segments. Its classical version allows querying which of the stored segments contain a given point, but our modification allows us to perform (query) any binary operation on any range in the array in O(logN) time. Here, we have used addition (+). For range updates, we have used lazy propagation.

• Space Complexity : O(NlogN)
  
• Build Time Complexity : O(NlogN)
  
• Query Time Complexity : O(logN)
  

  Author

  Madhav Gaba

  Soham Roy

Definition in file segtree.cpp.

Function Documentation

ConsTree()

void ConsTree	(	const std::vector< int64_t > &	arr,
		std::vector< int64_t > *	segtree,
		uint64_t	low,
		uint64_t	high,
		uint64_t	pos )

for std::vector

for assert for log2 for std::uint64_t for IO operations

Constructs the initial segment tree

Parameters

arr	input to construct the tree out of
segtree	the segment tree
low	inclusive lowest index of arr to begin at
high	inclusive highest index of arr to end at
pos	index of segtree to fill (eg. root node)

Returns

void

Definition at line 38 of file segtree.cpp.

                                                         {
    if (low == high) {
        (*segtree)[pos] = arr[low];
        return;
    }
 
    uint64_t mid = (low + high) / 2;
    ConsTree(arr, segtree, low, mid, 2 * pos + 1);
    ConsTree(arr, segtree, mid + 1, high, 2 * pos + 2);
    (*segtree)[pos] = (*segtree)[2 * pos + 1] + (*segtree)[2 * pos + 2];
}

main()

int main

(

void

)

Main function.

Returns

0 on exit

Definition at line 168 of file segtree.cpp.

           {
    test();  // run self-test implementations
 
    std::cout << "Enter number of elements: ";
 
    uint64_t n = 0;
    std::cin >> n;
 
    auto max = static_cast<uint64_t>(2 * pow(2, ceil(log2(n))) - 1);
    std::vector<int64_t> arr(n), lazy(max), segtree(max);
 
    int choice = 0;
    std::cout << "\nDo you wish to enter each number?:\n"
                 "1: Yes\n"
                 "0: No (default initialize them to 0)\n";
 
    std::cin >> choice;
    if (choice == 1) {
        std::cout << "Enter " << n << " numbers:\n";
        for (int i = 1; i <= n; i++) {
            std::cout << i << ": ";
            std::cin >> arr[i];
        }
    }
 
    ConsTree(arr, &segtree, 0, n - 1, 0);
 
    do {
        std::cout << "\nMake your choice:\n"
                     "1: Range update (input)\n"
                     "2: Range query (output)\n"
                     "0: Exit\n";
        std::cin >> choice;
 
        if (choice == 1) {
            std::cout << "Enter 1-indexed lower bound, upper bound & value:\n";
 
            uint64_t p = 1, q = 1, v = 0;
            std::cin >> p >> q >> v;
            update(&segtree, &lazy, p - 1, q - 1, v, 0, n - 1, 0);
        } else if (choice == 2) {
            std::cout << "Enter 1-indexed lower bound & upper bound:\n";
 
            uint64_t p = 1, q = 1;
            std::cin >> p >> q;
            std::cout << query(&segtree, &lazy, p - 1, q - 1, 0, n - 1, 0);
            std::cout << "\n";
        }
    } while (choice > 0);
 
    return 0;
}

query()

int64_t query	(	std::vector< int64_t > *	segtree,
		std::vector< int64_t > *	lazy,
		uint64_t	qlow,
		uint64_t	qhigh,
		uint64_t	low,
		uint64_t	high,
		uint64_t	pos )

Returns the sum of all elements in a range.

Parameters

segtree	the segment tree
lazy	for lazy propagation
qlow	lower index of the required query
qhigh	higher index of the required query
low	lower index of query for this function call
high	higher index of query for this function call
pos	index of segtree to consider (eg. root node)

Returns

result of the range query for this function call

Definition at line 63 of file segtree.cpp.

                            {
    if (low > high || qlow > high || low > qhigh) {
        return 0;
    }
 
    if ((*lazy)[pos] != 0) {
        (*segtree)[pos] += (*lazy)[pos] * (high - low + 1);
 
        if (low != high) {
            (*lazy)[2 * pos + 1] += (*lazy)[pos];
            (*lazy)[2 * pos + 2] += (*lazy)[pos];
        }
        (*lazy)[pos] = 0;
    }
 
    if (qlow <= low && qhigh >= high) {
        return (*segtree)[pos];
    }
 
    uint64_t mid = (low + high) / 2;
 
    return query(segtree, lazy, qlow, qhigh, low, mid, 2 * pos + 1) +
           query(segtree, lazy, qlow, qhigh, mid + 1, high, 2 * pos + 2);
}

test()

static void test ( )

static

Self-test implementation.

Returns

void

Definition at line 147 of file segtree.cpp.

                   {
    auto max = static_cast<int64_t>(2 * pow(2, ceil(log2(7))) - 1);
    assert(max == 15);
 
    std::vector<int64_t> arr{1, 2, 3, 4, 5, 6, 7}, lazy(max), segtree(max);
    ConsTree(arr, &segtree, 0, 7 - 1, 0);
 
    assert(query(&segtree, &lazy, 1, 5, 0, 7 - 1, 0) == 2 + 3 + 4 + 5 + 6);
 
    update(&segtree, &lazy, 2, 4, 1, 0, 7 - 1, 0);
    assert(query(&segtree, &lazy, 1, 5, 0, 7 - 1, 0) == 2 + 4 + 5 + 6 + 6);
 
    update(&segtree, &lazy, 0, 6, -2, 0, 7 - 1, 0);
    assert(query(&segtree, &lazy, 0, 4, 0, 7 - 1, 0) == -1 + 0 + 2 + 3 + 4);
}

update()

void update	(	std::vector< int64_t > *	segtree,
		std::vector< int64_t > *	lazy,
		int64_t	start,
		int64_t	end,
		int64_t	delta,
		uint64_t	low,
		uint64_t	high,
		uint64_t	pos )

Updates a range of the segment tree.

Parameters

segtree	the segment tree
lazy	for lazy propagation
start	lower index of the required query
end	higher index of the required query
delta	integer to add to each element of the range
low	lower index of query for this function call
high	higher index of query for this function call
pos	index of segtree to consider (eg. root node)

Returns

void

Definition at line 103 of file segtree.cpp.

                                         {
    if (low > high) {
        return;
    }
 
    if ((*lazy)[pos] != 0) {
        (*segtree)[pos] += (*lazy)[pos] * (high - low + 1);
 
        if (low != high) {
            (*lazy)[2 * pos + 1] += (*lazy)[pos];
            (*lazy)[2 * pos + 2] += (*lazy)[pos];
        }
        (*lazy)[pos] = 0;
    }
 
    if (start > high || end < low) {
        return;
    }
 
    if (start <= low && end >= high) {
        (*segtree)[pos] += delta * (high - low + 1);
 
        if (low != high) {
            (*lazy)[2 * pos + 1] += delta;
            (*lazy)[2 * pos + 2] += delta;
        }
 
        return;
    }
 
    uint64_t mid = (low + high) / 2;
 
    update(segtree, lazy, start, end, delta, low, mid, 2 * pos + 1);
    update(segtree, lazy, start, end, delta, mid + 1, high, 2 * pos + 2);
    (*segtree)[pos] = (*segtree)[2 * pos + 1] + (*segtree)[2 * pos + 2];
}