Segment trees for range queries in Data Structures Theory - Time & Space Complexity

Time Complexity: Segment trees for range queries

O(log n)

Understanding Time Complexity

When using segment trees for range queries, it is important to understand how the time to answer queries grows as the input size increases.

We want to know how fast we can get answers when the data size gets bigger.

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

function rangeQuery(tree, n, left, right) {
  left += n;  // shift index to leaf
  right += n;
  let result = 0;
  while (left <= right) {
    if ((left % 2) === 1) result += tree[left++];
    if ((right % 2) === 0) result += tree[right--];
    left = Math.floor(left / 2);
    right = Math.floor(right / 2);
  }
  return result;
}
    

This code performs a range sum query on a segment tree represented as an array.

Identify Repeating Operations

• Primary operation: The while loop that moves the left and right pointers up the tree.
• How many times: The loop runs roughly twice the height of the tree, which depends on the input size.

How Execution Grows With Input

As the input size grows, the height of the segment tree grows slowly, so the number of steps in the loop grows slowly too.

Input Size (n)	Approx. Operations
10	About 8 steps
100	About 14 steps
1000	About 20 steps

Pattern observation: The number of steps grows slowly, roughly with the height of the tree, which increases as the input size doubles.

Final Time Complexity

Time Complexity: O(log n)

This means the time to answer a range query grows slowly, only a little more work is needed even if the input size gets much bigger.

Common Mistake

[X] Wrong: "The range query will check every element between left and right, so it takes linear time."

[OK] Correct: The segment tree uses a clever structure to jump over parts of the range, so it does not check every element one by one.

Interview Connect

Understanding how segment trees speed up range queries shows you can work with smart data structures that handle big data efficiently.

Self-Check

"What if we changed the segment tree to support range updates as well as queries? How would the time complexity change?"

Practice

1. What is the main purpose of a segment tree in data structures?

easy

A. To quickly answer queries about a range of elements in an array

B. To sort elements in ascending order

C. To store elements in a linked list format

D. To perform binary search on a sorted array

Solution

1. Step 1: Understand the role of segment trees

   Segment trees are designed to efficiently answer queries about ranges, such as sums or minimums, in an array.

2. Step 2: Compare with other data structures

   Unlike sorting or binary search, segment trees focus on range queries and updates, not ordering or searching single elements.

3. Final Answer:

   To quickly answer queries about a range of elements in an array -> Option A

4. Quick Check:

   Segment tree = fast range queries [OK]

Hint: Segment trees speed up range queries, not sorting or searching [OK]

Common Mistakes:

• Confusing segment trees with sorting algorithms
• Thinking segment trees are for searching single elements
• Assuming segment trees store data like linked lists

2. Which of the following is the correct way to represent a segment tree for an array of size n?

easy

A. Using an array of size 2*n - 1

B. Using a linked list of size n

C. Using an array of size n

D. Using a binary search tree with n nodes

Solution

1. Step 1: Recall segment tree size formula

   A segment tree for an array of size n is usually stored in an array of size about 2*n - 1 to hold all nodes including internal and leaf nodes.

2. Step 2: Eliminate incorrect options

   Linked lists and binary search trees do not represent segment trees efficiently. An array of size n is too small to hold all nodes.

3. Final Answer:

   Using an array of size 2*n - 1 -> Option A

4. Quick Check:

   Segment tree size ≈ 2*n - 1 [OK]

Hint: Segment tree array size is about twice the input size minus one [OK]

Common Mistakes:

• Using linked lists instead of arrays
• Assuming segment tree size equals input size
• Confusing segment trees with binary search trees

3. Given the array [2, 5, 1, 4], what is the result of a range sum query from index 1 to 3 (0-based) using a segment tree?

medium

A. 6

B. 9

C. 10

D. 7

Solution

1. Step 1: Identify the elements in the range

   Indices 1 to 3 correspond to elements 5, 1, and 4.

2. Step 2: Calculate the sum of these elements

   Sum = 5 + 1 + 4 = 10.

3. Step 3: Verify the options

   10 is 10, but the question asks for the sum from index 1 to 3 inclusive, which is 5 + 1 + 4 = 10.

4. Final Answer:

   10 -> Option C

5. Quick Check:

   Sum(5,1,4) = 10 [OK]

Hint: Add elements in the given index range carefully [OK]

Common Mistakes:

• Off-by-one errors in index range
• Adding wrong elements outside the range
• Confusing 0-based and 1-based indexing

4. Consider a segment tree built for minimum queries on the array [3, 8, 7, 6, 2]. If the update operation changes the element at index 2 to 1, what is the new minimum value in the range 1 to 4?

medium

A. 2

B. 6

C. 3

D. 1

Solution

1. Step 1: Understand the update effect

   Original array: [3, 8, 7, 6, 2]. After update at index 2: [3, 8, 1, 6, 2].

2. Step 2: Find minimum in range 1 to 4

   Elements in range: 8, 1, 6, 2. Minimum is 1.

3. Final Answer:

   1 -> Option D

4. Quick Check:

   Updated min in range = 1 [OK]

Hint: Update changes minimum; re-check range after update [OK]

Common Mistakes:

• Forgetting to update the segment tree after change
• Using old values for minimum query
• Misidentifying the range boundaries

5. You have a large array and need to perform many range sum queries and point updates efficiently. Which of the following best explains why a segment tree is preferred over a simple prefix sum array?

hard

A. Segment trees use less memory than prefix sums

B. Segment trees allow both updates and queries in logarithmic time, prefix sums require linear time for updates

C. Prefix sums allow faster updates than segment trees

D. Prefix sums can handle range minimum queries better than segment trees

Solution

1. Step 1: Compare update and query times

   Prefix sums allow fast range sum queries but updating an element requires recalculating many prefix sums, which is linear time.

2. Step 2: Understand segment tree advantages

   Segment trees support both point updates and range queries in O(log n) time, making them efficient for many operations.

3. Final Answer:

   Segment trees allow both updates and queries in logarithmic time, prefix sums require linear time for updates -> Option B

4. Quick Check:

   Segment tree = fast updates + queries [OK]

Hint: Segment trees handle updates and queries fast; prefix sums don't [OK]

Common Mistakes:

• Assuming prefix sums update quickly
• Thinking segment trees use less memory
• Confusing range sum with range minimum queries