We're live on Product Hunt today! Use code PRODUCTHUNT26 for 20% off. Upvote us ↑
DSA Trainer
← 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 both full courses, DSA and System Design, 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 both full courses, DSA and System Design, 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 DSA course: concept lessons, drills, graded tests, and cold-read capstones for every pattern
  • The full System Design course: 20 building-block units for the design round, included at no extra cost
  • 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.

Full access · both courses

Full access

$49 / year

or $9 / month

Now includes a C++ runner alongside JavaScript and Python.

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.

  • Both courses, every unit: DSA + System Design
  • 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.