数组中的第k个最大元素

问题描述：

给定一个整数数组和一个整数k，请返回数组中第k个最大的元素。

思路：

这一题重要的是排序。

即如果我们按照从小到大排序，那么返回数组第n-k位置的元素。

如果我们按照从大到小排序，那么返回数组第k-1位置的元素。

在这里使用两种排序方法。

1.快速排序：

代码如下：

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

    int findKthLargest(vector<int>& nums, int k) {
        int n=nums.size();
        return quickselect(nums,0,n-1,n-k);
    }
};

在快排函数中添加了一个参数k，最后直接返回该位置的元素。其实主要还是快速排序的实现。

2.堆排序

代码如下：

class Solution {
public:
    void maxHeapify(vector<int>& a, int i, int heapSize) {
        int l = i * 2 + 1, r = i * 2 + 2, largest = i;
        if (l < heapSize && a[l] > a[largest]) {
            largest = l;
        } 
        if (r < heapSize && a[r] > a[largest]) {
            largest = r;
        }
        if (largest != i) {
            swap(a[i], a[largest]);
            maxHeapify(a, largest, heapSize);
        }
    }

    void buildMaxHeap(vector<int>& a, int heapSize) {
        for (int i = heapSize / 2; i >= 0; --i) {
            maxHeapify(a, i, heapSize);
        } 
    }

    int findKthLargest(vector<int>& nums, int k) {
        int heapSize = nums.size();
        buildMaxHeap(nums, heapSize);
        for (int i = nums.size() - 1; i >= nums.size() - k + 1; --i) {
            swap(nums[0], nums[i]);
            --heapSize;
            maxHeapify(nums, 0, heapSize);
        }
        return nums[0];
    }
};