力扣215. 数组中的第K个最大元素

 1.二叉最小堆法：维护一个size为k的最小堆，每次从堆中去除一个比item更小的元素，最后留下的便是最大的k个元素。（nlogn)

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        priority_queue<int, vector<int>, greater<int>> heap;
        for (int item : nums) {
            heap.push(item);
            if (heap.size() > k) {
                heap.pop();
            }
        }
        return heap.top();
    }
};

 2.快速选择法：

为了解决存在相同元素的问题，需要注意

1. 必须使用do while

// 假设数组: [3, 3, 3, 3], pivot=3
while(nums[i] < p) {++i;} // 永远不会进入，i停留在l
while(nums[j] > p) {--j;} // 永远不会进入，j停留在r
// i和j都不移动，死循环！

2.

 

class Solution {
public:
    int partition(vector<int>& nums, int l, int r, int k) {
        if (l == r) {
            return nums[l];
        }
        int p = nums[l];
        int i = l - 1, j = r + 1;
        while(i < j) {
            do i++; while(nums[i] < p);
            do j--; while(nums[j] > p);
            if (i < j)
                swap(nums[i], nums[j]);
        }
        if (k > j) {
            return partition(nums, j + 1, r, k);
        } else {
            return partition(nums, l, j, k);
        }
    }
    int findKthLargest(vector<int>& nums, int k) {
        return partition(nums, 0, nums.size() - 1, nums.size() - k);

    }
};