给定一个最大堆,找出第k大的节点。
建一个最小堆 minHeap来解决此题.
1 取最大堆的根节点,放入minHeap;
2 移除minHeap的根节点,并把该根节点在最大堆的两个子节点插入minHeap(插入过程保证heap的性质);
3 回到1.2 步。
minHeap的节点数在不断增加,但是不会超过2k。minHeap插入节点的复杂度是O(2k)=O(k)。整个算法的复杂度是kO(k)。
类似题目:
1. 有序矩阵中查找第k小的元素 Kth smallest element in a row-wise and column-wise sorted 2D array