diff --git a/heaps/median_from_stream.py b/heaps/median_from_stream.py deleted file mode 100644 index 34314c3..0000000 --- a/heaps/median_from_stream.py +++ /dev/null @@ -1,37 +0,0 @@ -#!/usr/bin/env python3 -# -*- coding: utf-8 -*- -# author: bt3gl - - -class MedianFinder: - - def __init__(self): - - self.min_heap = [] - self.max_heap = [] - - - def addNum(self, num: int) -> None: - - min_heap, max_heap = self.min_heap, self.max_heap - - if len(min_heap) < len(max_heap): - heappush(min_heap, num) - - else: - heappush(max_heap, -num) - - if max_heap and min_heap and -max_heap[0] > min_heap[0]: - - heappush(max_heap, -heappop(min_heap)) - heappush(min_heap, -heappop(max_heap)) - - - def findMedian(self): - - min_heap, max_heap = self.min_heap, self.max_heap - - if len(min_heap) < len(max_heap): - return -max_heap[0] - - return (min_heap[0] - max_heap[0]) / 2