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];  }};