实现语言:Python3.9
题目来源:牛客

实现步骤:
- 1、使用堆的方式实现,具体实现思路:我们把数据从中点位置分为两个部分,前一部分构建成大顶堆A,后一部分构建成小顶堆B(注意前半部分的数据小于后半部分的数据);
- 2、先将数据添加到大顶堆中,然后取出大顶堆的堆顶元素,把堆顶元素添加到小顶堆中,这样做的目的是保证大顶堆中的数据全部小于小顶堆中的数据;
- 3、然后平衡A和B的数据,如果A中的数据个数小于B那么就把B的堆顶元素取出放入到A中,总之就是需要保证len(A)>=len(B);
- 4、这样如果len(A) > len(B),那么中点就是A部分的堆顶数据,如果len(A) == len(B) 那么中点就是A和B的堆顶数据和的一半。
import heapq
class Solution:
    min_heap = []
    max_heap = []
    # 构建大顶堆
    def my_max_heapify(self, it):
        max_heap_ = [(-x, x) for x in it]
        heapq.heapify(max_heap_)
        return [x for (_, x) in max_heap_]
    def Insert(self, num):
        # write code here
        # 先加入大顶堆中
        self.max_heap.append(num)
        # 构建大顶堆
        self.max_heap = self.my_max_heapify(self.max_heap)
        # 将大顶堆中的堆顶元素取出,放到小顶堆中
        heapq.heappush(self.min_heap, heapq.heappop(self.max_heap))
        # 再次构建大顶堆
        self.max_heap = self.my_max_heapify(self.max_heap)
        # 平衡两个堆中的元素数量
        # 如果大顶堆中的元素比小顶堆中的元素少,则将小顶堆的堆顶元素取出放到大顶堆。
        if len(self.max_heap) < len(self.min_heap):
            self.max_heap.append(heapq.heappop(self.min_heap))
            self.max_heap = self.my_max_heapify(self.max_heap)
        
    def GetMedian(self):
        # write code here
        # 如果大顶堆的元素比小顶堆的元素个数多,说明大顶堆的堆顶元素就是中间值
        if len(self.max_heap) > len(self.min_heap):
            middle = self.max_heap[0]
            return middle
        # 如果相等,则中间元素是大顶堆和小顶堆的堆顶元素
        else:
            a =  self.max_heap[0]
            b =  self.min_heap[0]
            return (a + b) / 2


















