Search⌘ K
AI Features

Problem: Find Median from Data Stream

Explore how to design a MedianOfStream class that maintains a dynamic list of integers and efficiently finds the median in constant time. Learn to implement a dual-heap approach using max and min heaps to balance the data stream and retrieve the median quickly while managing insertions and rebalancing.

Statement

Design a data structure that stores a dynamically changing list of integers and can find the median in constant time, O(1)O(1), as the list grows. To do that, implement a class named MedianOfStream with the following functionality:

  • Constructor(): Initializes an instance of the class.

  • insertNum(int num): Adds a new integer num to the data structure.

  • findMedian(): Returns the median of all integers added so far.

Note: The median is the middle value in a sorted list of integers.

  • For an odd-sized list (e.g., [4,5,6][4, 5, 6]), the median is the middle element: 55.

  • For an even-sized list (e.g., [2,4,6,8][2, 4, 6, 8]), the median is the average of the two middle elements: (4+6)/2=5 ...