Prefix Sums & Range Queries · 12 min · free with an account

2D Prefix Sum: When to Use It

Lesson preview · about 30%

Recognise the problem shape before choosing 2D Prefix Sum.

Core idea

Choose a pattern from the input guarantees and required operations, not from surface wording.

Start with the idea

A problem is a good match for 2D Prefix Sum when it asks for image regions, matrix queries, grid density. Here the concrete task is: Return the sum of rows 1..2 and columns 0..1. The visible input [[1,2,3],[4,5,6],[7,8,9]] is an example, not a definition of the whole problem. Ask what operations the solution must support and what the input guarantees. Precomputation pays when many ranges are queried. If values change between queries, an updateable structure may be needed; for one small query, a direct walk is simpler. The right choice follows from those guarantees, not from a familiar noun in the prompt.

The decisive invariant is: Store each rectangle from the origin and use inclusion-exclusion for any subrectangle. Try to say why it applies to the task without quoting the implementation. Which fact becomes known after one update, and which possible answers remain? A range method summarizes intervals so the same information is not recomputed. A static prefix table and an updateable tree make different tradeoffs; define endpoint conventions first. If that claim cannot be maintained on a legal input, the pattern is wrong or one of its preconditions is missing. A counterexample is more useful than a vague warning: change one property of the sample, trace the first two steps, and identify the exact claim that fails.

Compare costs before committing. The advertised bound is O(rows*cols) build · O(1) query; identify the operation responsible for it and the extra memory it needs.

Keep learning, for free

The rest of this lesson is waiting for you.

Create a free account to read the complete explanation, see the infographic, explore the code, and try the exercises. All 742 lessons and algorithm steps are included.

No payment details. Your account opens the complete Academy.

← All lessons in Algorithms