116：Kth Largest Element in an Array

题目：Find the k -th largest element in an unsorted array. For example, given [3,2,1,5,6,4] and k = 2 , return 5.
Note: You may assume k is always valid, 1 ≤ k ≤ array’s length.

解析1：堆排序，维护一个 k 大小的小根堆，将数组中的每个元素与堆顶元素进行比较，如果比堆顶元素大，则删除堆顶元素并添加该元素，如果比堆顶元素小，则什么也不做，继续下一个元素。时间复杂度 O(nlogk) ，空间复杂度 O(k)。代码如下：

// 时间复杂度为 O(nlogk)，空间复杂度为O(k)
int findKthLargest(vector<int>& array, int k){
        // c++ 中的堆默认是最大堆
        // 为了实现一个最小堆，需要用greater<int> [在头文件<functional>中]
        // 在此题中，我们需要使用最小堆
        priority_queue<int, vector<int>, greater<int>> pq;

        for (int i = 0; i < array.size(); ++i) {
                if (pq.size() < k)
                        pq.push(array[i]);
                else {
                        if (pq.top() < array[i]) {
                                pq.pop();
                                pq.push(array[i]);
                        }
                }
        }
        return pq.top();
}

解析2：利用快速排序中的partition思想，从数组中随机选择一个元素x，把数组划分为前后两部分 sa 和 sb ， sa 中的元素小于x， sb 中的元素大于或等于x。
这时有两种情况：
i. sa 的元素个数小于 k ，则递归求解 sb 中的第 k-|sa| 大的元素
ii. sa 的元素个数大于或等于 k ，则递归求解 sa 中的第 k 大的元素时间复杂度 O(n) ，空间复杂度 O(1)。
代码如下：

// 时间复杂度 O(n)，空间复杂度 O(1)
int findKthLargest(vector<int>& array, int k) {
        quickSelect(array, array.size() - k + 1, 0, array.size() - 1);
        return array[array.size() - k];
}

void quickSelect(vector<int>& array, int k, int left, int right) {
        if (left < right) {
                int breakPoint = randomPartition(array, left, right);
                if (breakPoint == k - 1)
                        return;
                else if (breakPoint < k - 1)
                        quickSelect(array, k, breakPoint + 1, right);
                else
                        quickSelect(array, k, left, breakPoint - 1);
        }
}
int randomPartition(vector<int>& array, int left, int right) {
        int pivotPosition = rand() % (right - left + 1) + left;
        swap(array[pivotPosition], array[right]);
        return partition(array, left, right);
}

int partition(vector<int>& array, int left, int right) {
        int compareVal =  array[right];

        int i = left - 1;
        for (int j = left; j != right; ++j)
                if (array[j] <= compareVal) {
                        ++i;
                        swap(array[i], array[j]);
                }
        swap(array[i + 1], array[right]);
        return i + 1;
}