215. Kth Largest Element in an Array

215. Kth Largest Element in an Array

---

Find the kth largest element in an unsorted array. Note that it is the kth largest element in the sorted order, not the kth distinct element.

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.

题目大意

从一个未经排序的数组中找出第k大的元素。

解题思路

使用快速排序的思想，将每次通过partition得到的位置与k比较，如果该位置与k相等，则直接结束排序；如果该位置比k大，则往左半部分找；如果该位置比k小，则往右半部分找。

算法复杂度

O(nlogn)

代码实现

class Solution {
public:
  int __partition(vector<int>& nums, int l, int r) {
    // 返回p, 使得nums[l, ... , p-1] > p, 且p > nums[0, ... , r]
    int v = nums[l];
    int p = l;
    for (int i = l+1; i <= r; ++i) {
      if (nums[i] > v) {
        // 将大于nums[l]的交换到p位置的左边
        swap(nums[i], nums[++p]);
      }
    }
    // 将nums[l]将换到目标位置p
    swap(nums[l], nums[p]);
    return p;
  }
  void __quickSort(vector<int>& nums, int l, int r, const int& k) {
    if (l >= r)
      return;

    int p = __partition(nums, l, r);
    if (p == k) {
      return;
    } else if (p > k) {
      // 左侧搜寻
      __quickSort(nums, l, p-1, k);
    } else {
      // 右侧搜寻
      __quickSort(nums, p+1, r, k);
    }
  }
  int findKthLargest(vector<int>& nums, int k) {
    __quickSort(nums, 0, nums.size()-1, k-1);  // 第k大 <=> 第k-1个位置
    return nums[k-1];
  }
};