数组中第K个最大元素

数组中第K个最大元素_第1张图片

维护一个K大小的最小堆,堆中元素个数小于K时,新元素直接进入堆;否则,当堆顶小于新元素时,弹出堆顶,将新元素加入堆。

由于堆是最小堆,堆顶是堆中最小元素,堆中K个元素就是已扫描的元素中最大的K个数,所以堆顶是第K大的数

设数组N个元素,求第K大的,时间复杂度 N*logK

class Solution {
public:
    int findKthLargest(vector& nums, int k) {
        priority_queue,greater > heap;//最小堆
        for(int i =0;iheap.top())
                {
                    heap.pop();
                    heap.push(nums[i]);
                }
            }
        }
        return heap.top();
    }
};

你可能感兴趣的:(LeetCode)