Part I · Arrays, Strings and Pointers Pattern 2 4 problems

Two Pointers

Sorting gives you a direction. Two pointers converging from the ends turn that direction into an O(n) search over all n² pairs.

The sliding window walks two indices forward together. Two pointers, in the classic sense, sends them toward each other from opposite ends. Each step throws away a whole family of candidate pairs at once, which is why a doubly-nested loop collapses into a single pass.

Contents

  1. When to use
  2. Core idea
  3. The three templates
  4. Common mistakes
  5. Two Sum II — Input Array Is Sorted
  6. Remove Duplicates from Sorted Array
  7. 3Sum
  8. Container With Most Water
  9. Recap

When to use

The trigger. The data is sorted, or can be sorted without changing the answer, and you need a pair, a triplet, or a rearrangement in place. The tell-tale phrases are “sorted array”, “find two numbers that…”, “O(1) extra space”, and “modify the array in place”.
SignalWhat it looks like
SortednessThe array is given sorted, or the answer does not depend on order so you may sort it yourself.
A pairwise relation“two numbers summing to…”, “the widest pair”, “closest to a target”.
Space pressure“in place”, “O(1) extra memory”, “do not allocate a new array”.
PalindromesAnything comparing a string to itself reversed is the same converging motion.

When not to use it

Core idea

Put left at index 0 and right at index n-1. Evaluate the pair. Because the array is sorted, the comparison tells you which pointer cannot possibly be part of the answer with any remaining partner, so you move that one inward and never look at it again. Each step eliminates an entire row or column of the n×n pair grid.
target = 13 1 3 4 6 8 9 left right 1 + 9 = 10 < 13 9 is the largest partner 1 can ever get so discard 1 entirely, move left in 1 3 4 6 8 9 3 + 9 = 12, still small → move left again
Figure 2.1 — One comparison removes five candidate pairs. That is where the factor of n goes.

Why discarding is safe

This is the proof to have ready, because it is the only interesting thing to say about the pattern.

Exchange argument. Suppose nums[left] + nums[right] < target. Since the array is sorted, nums[right] is the largest value still available. So nums[left] paired with anything left in the range gives a sum no bigger than the one just computed, which is already too small. Therefore nums[left] is in no solution, and dropping it loses nothing. The symmetric argument covers the too-large case.

Each iteration moves exactly one pointer inward by one, and the pointers start n-1 apart, so the loop runs at most n-1 times. O(n) after the sort.

The three templates

Template A — converging from both ends
def converge(nums: list[int], target: int) -> tuple[int, int] | None:
    """Find a pair in a sorted array summing to target."""
    # 0: first index. len(nums) - 1: last index, since indices start at 0.
    left, right = 0, len(nums) - 1

    # Invariant: no pair that uses an index outside left..right can sum to target.
    while left < right:                 # strict: never pair an element with itself
        total = nums[left] + nums[right]
        if total == target:
            return left, right
        if total < target:
            # +1: nums[left] is too small even with the biggest partner. Drop it.
            left += 1                   # need a bigger sum
        else:
            # -1: nums[right] is too big even with the smallest partner. Drop it.
            right -= 1                  # need a smaller sum

    return None                         # the pointers met: no pair exists

Use left < right, not <=, whenever the two pointers must select two different elements.

Template B — read and write, same direction
def compact(nums: list[int]) -> int:
    """Filter in place. Returns the length of the kept prefix."""
    # keep(x): placeholder test that says "this value stays".
    # Remove Duplicates: keep x if it differs from the last value kept.
    write = 0                           # 0: nothing kept yet. Next kept goes in slot 0

    # read: index of the value being judged.
    # Invariant: nums[0..write-1] holds every kept value from nums[0..read-1].
    for read in range(len(nums)):
        if keep(nums[read]):
            nums[write] = nums[read]    # write never overtakes read, so this is safe
            write += 1                  # +1: the kept prefix grew by one slot

    return write                        # write = count of kept values

The in-place filter. write <= read always holds, so the write can never clobber an element that has not been read yet.

keep rule: the value differs from the last kept value read = 3 1 2 2 3 3 write read 3 differs from nums[write − 1] = 2 so copy it into slot write after copy 1 2 3 3 3 write read nums[2] = 3, then write = 3 read moves on to index 4 read = 4 1 2 3 3 3 write read 3 equals nums[write − 1], so skip it return write = 3
Figure 2.2 — The write pointer never passes the read pointer, so the kept prefix grows safely in place.

Reading the figure. Green cells are the kept prefix, nums[0..write-1]. Amber is the value being read. Red is a value that is skipped. Grey cells past write hold leftovers that no longer matter.

Template C — fix one, converge on the rest
def triplets(nums: list[int], target: int) -> list[list[int]]:
    """Reduce a 3-sum to n independent 2-sums."""
    nums.sort()                         # sorted order is what lets Template A work
    out: list[list[int]] = []

    # i: index of the anchor. - 2: leave room for two partners after i.
    for i in range(len(nums) - 2):
        # i > 0: slot 0 has no left neighbour to compare. i - 1: the previous anchor.
        if i > 0 and nums[i] == nums[i - 1]:
            continue                    # skip duplicate anchors
        # i + 1: partners come after the anchor. len(nums) - 1: the last index.
        left, right = i + 1, len(nums) - 1
        # ... Template A on nums[left:right+1] for target - nums[i]
        # (right+1 because a slice end is exclusive, so right stays included)

    return out

k-sum is a for loop wrapped around (k-1)-sum. 3Sum is O(n²), 4Sum is O(n³).

fix nums[i], then run Template A on nums[i+1:] i = 1 −4 −1 −1 0 1 2 i left → ← right anchor −1: find a pair summing to 1 −1 + 2 = 1: record [−1, −1, 2] i = 2 −4 −1 −1 0 1 2 i nums[2] == nums[1], so skip this anchor it would find the same triplets again
Figure 2.3 — Template C turns 3Sum into one two-pointer pass per anchor, skipping repeated anchors.

Reading the figure. Violet is the fixed anchor i. Amber cells are left and right. Blue cells are still in range. Red is an anchor skipped because it repeats the one before.

Common mistakes

The problems

1. Two Sum II — Input Array Is Sorted Easy

Problem

Given a 1-indexed array numbers sorted in non-decreasing order, find the two numbers that add up to target and return their 1-based indices. Exactly one solution exists, and you may not use an element twice. Extra space must be O(1).

Approach

Solution

def two_sum_sorted(numbers: list[int], target: int) -> tuple[int, int]:
    """Indices of the two values in a sorted array that sum to target.

    Args:
        numbers: Non-decreasing integers.
        target: The required sum.

    Returns:
        The 1-based indices (i, j) with i < j and numbers[i-1] + numbers[j-1] == target.

    Raises:
        ValueError: If no such pair exists.

    Example:
        >>> two_sum_sorted([2, 7, 11, 15], 9)
        (1, 2)
    """
    # 0: first index. len(numbers) - 1: last index, since indices start at 0.
    left, right = 0, len(numbers) - 1

    # Invariant: the answer pair, if any, lies inside left..right.
    while left < right:                # strict: an element cannot pair with itself
        total = numbers[left] + numbers[right]

        if total == target:
            # + 1: the problem wants 1-based indices, Python uses 0-based.
            return left + 1, right + 1
        if total < target:
            left += 1        # numbers[left] is too small even with the largest partner
        else:
            # numbers[right] is too large even with the smallest partner.
            right -= 1

    raise ValueError("no pair sums to target")

Walkthrough

numbers = [2, 7, 11, 15], target = 18:

leftrightsumaction
0317too small, left++
1322too large, right--
1218match, return (2, 3)
step 1 2 7 11 15 left right 2 + 15 = 17 < 18 too small: drop 2, left moves right step 2 2 7 11 15 left right 7 + 15 = 22 > 18 too large: drop 15, right moves left step 3 2 7 11 15 left right 7 + 11 = 18, a match return 1-based indices (2, 3)
Figure 2.4 — Each comparison drops one end for good, and the third step finds 7 + 11 = 18.

Reading the figure. Amber cells are the two pointers. Blue cells are still in range. Red cells are dropped, since no remaining partner can work with them. Green is the answer pair.

TimeO(n)SpaceO(1)

Edge cases to raise

Say this out loud: “The O(1) space constraint is the signal. Without it I would use a hash map in one pass; with it, sortedness lets me converge.”

2. Remove Duplicates from Sorted Array Easy

Problem

Given a sorted array nums, remove duplicates in place so each value appears once, keeping the relative order. Return the number of unique elements k. The first k slots of nums must hold the result; what follows does not matter.

Approach

Solution

def remove_duplicates(nums: list[int]) -> int:
    """Compact a sorted list in place so each value appears once.

    Args:
        nums: Sorted integers. Mutated in place.

    Returns:
        k, the number of unique values. nums[:k] holds them in order.

    Example:
        >>> data = [0, 0, 1, 1, 1, 2, 2, 3]
        >>> k = remove_duplicates(data)
        >>> (k, data[:k])
        (4, [0, 1, 2, 3])
    """
    if not nums:
        return 0                       # 0: an empty list has no unique values

    write = 1                      # nums[0] is always kept, so the next slot is 1

    # read: index of the value being judged. Start at 1: slot 0 is already kept.
    # Invariant: nums[0..write-1] are the unique values of nums[0..read-1].
    for read in range(1, len(nums)):
        # nums[write - 1] is the last value we decided to keep.
        # Sorted input means a new value only has to differ from that one.
        if nums[read] != nums[write - 1]:
            nums[write] = nums[read]   # copy the new value into the next free slot
            write += 1                 # +1: the kept prefix grew by one

    return write

Why compare against nums[write - 1] and not nums[read - 1]

Both work for this problem, because sorted duplicates are adjacent and the prefix mirrors the source. But nums[write - 1] is the honest expression of the intent: is this different from the last thing I kept. That version generalises without change to the common follow-up, allow each value at most twice, where you compare against nums[write - 2]. The read - 1 version does not generalise.

Walkthrough

nums = [1, 1, 2]:

readnums[read]nums[write-1]actionwrite
111equal, skip1
221write nums[1] = 22
read = 1 1 1 2 write, read nums[1] = 1 equals nums[write − 1] = 1 skip it. write stays 1 read = 2 1 1 2 write read 2 differs from nums[0] = 1 copy: nums[1] = 2, then write = 2 result 1 2 2 write return k = 2 slot 2 is a leftover and does not matter
Figure 2.5 — The duplicate 1 is skipped, the 2 is copied left, and the first two slots hold the answer.

Reading the figure. Green cells are the kept prefix. Amber is the value being copied. Red is a skipped duplicate. Grey cells are not kept yet or are leftovers.

TimeO(n)SpaceO(1)

The follow-up

def remove_duplicates_at_most_twice(nums: list[int]) -> int:
    """Same idea, but each value may survive twice."""
    write = 0                      # 0: nothing kept yet. Next kept goes in slot 0

    # value: the item being judged.
    # Invariant: nums[0..write-1] holds each value seen so far, at most twice.
    for value in nums:
        # Keep it unless the last two kept values are already this value.
        # write < 2: fewer than 2 kept, so no value can have 2 copies yet.
        # write - 2: two slots back. Sorted input means a match there also
        # matches the slot in between. So value would be a third copy.
        if write < 2 or value != nums[write - 2]:
            nums[write] = value
            write += 1                 # +1: the kept prefix grew by one

    return write

One changed constant. This is why the write - 1 framing is worth the habit.

Edge cases to raise

Say this out loud: “write never passes read, so writing into the prefix cannot destroy data I still need to read.”

3. 3Sum Medium

Problem

Given an array nums, return all unique triplets [a, b, c] with a + b + c == 0. The result must not contain duplicate triplets.

Approach

Solution

def three_sum(nums: list[int]) -> list[list[int]]:
    """All unique triplets from nums that sum to zero.

    Args:
        nums: Integers, in any order. Sorted in place as a side effect.

    Returns:
        A list of triplets, each sorted ascending, with no duplicates.

    Example:
        >>> three_sum([-1, 0, 1, 2, -1, -4])
        [[-1, -1, 2], [-1, 0, 1]]
    """
    nums.sort()                        # sorted order makes the two-pointer scan valid
    n = len(nums)
    triplets: list[list[int]] = []

    # i: index of the anchor. n - 2: leave room for two partners after i.
    for i in range(n - 2):
        # Sorted, so once the anchor is positive the three smallest remaining
        # values are all positive and no triplet can reach zero.
        if nums[i] > 0:                # 0: the target sum
            break
        # Dedup 1: an anchor value already used produces the same triplets.
        # i > 0: slot 0 has no earlier anchor. i - 1: the previous anchor.
        if i > 0 and nums[i] == nums[i - 1]:
            continue

        # i + 1: partners sit after the anchor. n - 1: the last index.
        left, right = i + 1, n - 1
        # Invariant: every new triplet for this anchor uses partners in left..right.
        while left < right:
            total = nums[i] + nums[left] + nums[right]

            if total < 0:              # 0: target. Too small, so raise the low end
                left += 1
            elif total > 0:            # too big, so lower the high end
                right -= 1
            else:
                triplets.append([nums[i], nums[left], nums[right]])
                left += 1              # +1: step past the low partner just used
                right -= 1             # -1: step past the high partner just used
                # Dedup 2 and 3: slide past repeats of the values just used.
                # left - 1: the value just used. Skip while it repeats.
                while left < right and nums[left] == nums[left - 1]:
                    left += 1
                # right + 1: the value just used. Skip while it repeats.
                while left < right and nums[right] == nums[right + 1]:
                    right -= 1

    return triplets

Walkthrough of the deduplication

nums = [-2, 0, 0, 2, 2] after sorting. With anchor -2 at i = 0, the pointers find (0, 2) and record [-2, 0, 2]. Both pointers move, landing on the second 0 and the second 2, which are repeats, so both dedup loops advance past them and the pointers cross. Without those loops the same triplet is recorded twice.

match −2 0 0 2 2 i left right −2 + 0 + 2 = 0: record [−2, 0, 2] then move both pointers inward both move −2 0 0 2 2 i left right left lands on a second 0 right lands on a second 2 dedup −2 0 0 2 2 i left, right the 0 repeats, so left skips to 3 left == right: the loop ends
Figure 2.6 — The skip loops step over repeated values, so the triplet [−2, 0, 2] is recorded once.

Reading the figure. Violet is the anchor i. Amber cells are left and right. Blue cells are still in range. Red is a repeat that the skip loop steps over.

Why not use a set of tuples instead

Collecting into a set of sorted tuples also produces the right answer and is easier to write under pressure. It costs extra memory and hashing, and it hides the fact that you understand where the duplicates come from. Write the explicit skips if you can; mention the set as the fallback you would use if short on time.
TimeO(n²)SpaceO(1) extraSortO(n log n)

The outer loop runs n times, each inner converge is O(n). Output space is not counted; the sort is in place.

Edge cases to raise

Say this out loud: “Sorting costs O(n log n) but the main loop is O(n²) anyway, so the sort is free, and it gives me both the converging scan and adjacent duplicates.”

4. Container With Most Water Medium

Problem

height[i] is the height of a vertical line at position i. Pick two lines so that the container they form with the x-axis holds the most water. Return that maximum area. The area for a pair is (j - i) * min(height[i], height[j]).

Approach

Solution

def max_area(height: list[int]) -> int:
    """Largest area of water trapped between two vertical lines.

    Args:
        height: Non-negative line heights, indexed by position.

    Returns:
        The maximum area. 0 if fewer than two lines.

    Example:
        >>> max_area([1, 8, 6, 2, 5, 4, 8, 3, 7])
        49
    """
    # 0: leftmost line. len(height) - 1: rightmost line. Widest pair first.
    left, right = 0, len(height) - 1
    best = 0                           # 0: no container measured yet

    # Invariant: every pair with an end outside left..right is no better than best.
    while left < right:
        span = right - left            # width between the two lines
        # The shorter line caps the water level.
        best = max(best, span * min(height[left], height[right]))

        # Only moving the shorter line can raise the min, so only that move
        # has any chance of beating the current area.
        if height[left] < height[right]:
            left += 1                  # +1: give up the short left line
        else:
            right -= 1                 # -1: give up the short right line

    return best

The correctness argument, stated properly

Suppose height[left] < height[right]. Consider any pair (left, j) with j < right. Its width is smaller than right - left, and its height is at most height[left], which is the current minimum. So its area is strictly less than the area just computed. Every remaining pair that uses left is therefore dominated, and discarding left cannot lose the optimum. That is exactly the exchange argument from the top of the page, with width in place of sortedness.

Walkthrough

height = [1, 8, 6, 2, 5, 4, 8, 3, 7]:

leftrightspanmin heightareabest
088188
18774949
17631849
16584049
left 0, right 8 8 × 1 = 8 best 8, move left left 1, right 8 7 × 7 = 49 best 49, move right left 1, right 7 6 × 3 = 18 best 49, move right left 1, right 6 5 × 8 = 40 best 49, move right
Figure 2.7 — Moving the shorter line each step finds the area 49 on the second step.

Reading the figure. Each chart is one step. Amber bars are left and right. Blue bars are still in range. The shaded box is the water, as wide as the span and as tall as the shorter line. Green marks the best area.

The scan continues but never beats 49.

TimeO(n)SpaceO(1)

Edge cases to raise

Do not confuse this with Trapping Rain Water. That problem asks how much water sits on top of a terrain profile, uses all the bars, and needs prefix maxima from both sides. It is also solvable with two pointers, but the state and the reasoning are different. Interviewers ask them back to back.
Say this out loud: “The area is limited by the shorter line, so moving the taller one strictly shrinks the answer. That makes moving the shorter line the only candidate move, and gives a one-pass solution.”

Recap

The six things to carry forward

Where this goes next

Pattern 3, Fast and Slow Pointers, keeps two pointers moving in the same direction but at different speeds. That difference in speed is what detects a cycle in a structure you cannot index into.


← 1 — Sliding Window 3 — Fast and Slow Pointers →