DSA Trainer
Heaps and Priority Queues
← All problems
PremiumTwo HeapsHard

Find Median from Data Stream

Asked at:AmazonGoogleMicrosoft

Design a data structure that supports adding integers from a data stream and retrieving the median at any time. Implement MedianFinder with addNum(num) and findMedian() methods.

This is a premium problem.

Unlock Find Median from Data Stream with the full guided learning path and every other premium problem.

Problem

The median is the middle value in an ordered integer list. Implement the MedianFinder class that supports addNum and findMedian operations.

Input

A series of addNum(num) calls followed by findMedian() calls.

Output

findMedian() returns the median of all numbers added so far. If there is an even count, return the average of the two middle values.

Examples

Input: addNum(1), addNum(2), findMedian(), addNum(3), findMedian()

Output: 1.5, 2.0

After [1,2]: median = (1+2)/2 = 1.5. After [1,2,3]: median = 2.

The brute-force approach

Keep a sorted list. On each addNum, insert in sorted position (O(n) shift). On findMedian, return the middle element.

def addNum(num):
    # binary search for position, then insert
    bisect.insort(data, num)

def findMedian():
    n = len(data)
    if n % 2 == 1: return data[n//2]
    return (data[n//2 - 1] + data[n//2]) / 2

Insertion into a sorted list costs O(n) due to the array shift. For n total numbers, total cost is O(n²).

Time: O(n) per addNum, O(1) findMedianSpace: O(n)

Spotting the pattern

This is a Two Heaps problem. The key question to ask yourself:

If you could instantly see both the largest number in the bottom half and the smallest number in the top half, what would the median be?

Answering that is where it clicks, and it's exactly what the guided walkthrough below builds with you: the pattern reasoning, a progressive hint ladder that never spoils the answer, a row-by-row dry run, the optimized solution, and an in-browser editor to run your code against real test cases.

Premium

Unlock the full guided solution

The Two Heaps walkthrough for Find Median from Data Stream: the progressive hint ladder, a row-by-row dry run, the optimized code, and an in-browser runner, plus the full guided learning path and every other premium problem.

Already finished Big-O and Hash Maps free? This picks up right where the path leads: same format, harder patterns, the ones that show up in every FAANG-style screen.

What you get

  • The full guided learning path: concept lessons, drills, graded tests, and cold-read capstones for every pattern
  • Stop going blank on algorithm problems. The hint ladder walks you to the answer without spoiling it.
  • Actually internalize the trace: dry runs walk state row by row so it clicks before you write a line
  • Know when to reach for each pattern. Every problem names the trigger so you spot it next time.
  • Write and run JS or Python in the browser with real test cases on every problem
  • Access every new problem, unit, and pattern guide as they ship weekly
B
Big_Wolverine_7575Founding Member· unprompted on Reddit

“This app finally made it click. I tried NeetCode, CTCI, and YouTube for years and still couldn't solve Two Sum. The beginner mental models for the patterns are genuinely so helpful. THANK YOU.

Price locks in for life · First 100 only

Founding Member Deal

$29 / year

or $6 / month

LeetCode Premium is $159/year and doesn't teach you anything.

Less than one Udemy course, with guided problems instead of passive lectures.

One extra month of job searching costs more than a full year here.

Your price is locked in for life.

When the 100th founding spot closes, the price goes up for everyone who joins after. Your rate never changes, no matter how much the curriculum grows.

  • The full guided learning path, every unit
  • All 150 problems across 33 patterns, plus every new one
  • New problems and units weekly
  • Cancel anytime

Cancel anytime. Not useful within 7 days? Email for a full refund. Secured by Stripe.