【28】215. Kth Largest Element in an Array

215. Kth Largest Element in an Array

• Total Accepted: 108089
• Total Submissions: 286063
• Difficulty: Medium
• Contributors: Admin

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.

Solution: 1 sort 2 partition - time limitation

class Solution {
public:
//最简单暴力的方法就是先对整个数组排序，然后返回 nums[k-1] 即可。时间复杂度O(nlogn)。
//采用快速排序的思路，但是不对整个数列排序。
//partition(int[] a, int lo, int hi)方法和快速排序一样，选取该分段中的第一个数为标兵(pivot)。所有大于pivot的数放到其左侧，小于pivot的数放到其右侧。最后返回pivot的位置。在经过一次partition方法之后，根据k和pivot的位置，来选择是对左侧部分还是对右侧部分继续进行partition。该方法时间复杂度为O(n)。
    int findKthLargest(vector<int>& nums, int k) {
        //sort(nums.begin(), nums.end());
        //return nums[nums.size() - k];
        k--;
        int l = 0, r = nums.size() - 1;
        int idx = 0;
        while(l < r){
            idx = partition(nums, l, r);
            if(k < idx) r = idx - 1;
            else if(k > idx) l = idx + 1;
            else return nums[idx];
        }
        
        return nums[l];
    }
    
    int partition(vector<int>& nums, int left, int right){
        int l = left;
        int r = right;
        int pivot = nums[l];
        while(l < r){
            while(l < r && nums[r] < pivot) r--;
            nums[l] = nums[r];
            while(l < r && nums[l] > pivot) l++;
            nums[r] = nums[l];
        }
        nums[l] = pivot;
        return l;
    }
};