Leetcode 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.

法一：

1、先排序（O（n*logN））

2、返回n-k个元素

 int findKthLargest(vector<int>& nums, int k) {
        //S1 sort函数 时间复杂度O(logN*N)
        int n = nums.size();
        std::sort(nums.begin(),nums.end());
        int result = nums[n-k];
        return result;
        }

法二：快速选择（随机选择）

1、同快排类似，但是每次划分只需要关注划分中的一段。

2、最差时间复杂度O（n*n），但出现可能性很低。

    时间复杂度的期望为O（n）

class Solution {
    
public:
    int findKthLargest(vector<int>& nums, int k) {
      
        //S5 quickSelect
        return quickSelect(nums,0,nums.size()-1,k);
        
    }
     int quickSelect(vector<int>& nums,int start,int end,int k){
         if(start == end) return nums[start];
         int pivot;
         pivot = randPartition(nums,start,end);
         if(end-pivot+1 == k) {return nums[pivot];}
         if(end-pivot+1 > k){
            return quickSelect(nums,pivot+1,end,k);
         }
         if(end-pivot+1 < k){
            return quickSelect(nums,start,pivot-1,pivot+k-end-1);
         }
    
         return -1;
     }
     int randPartition(vector<int>& nums,int start ,int end){
         int randi = std::rand()%(end-start+1)+start;
         std::swap(nums[randi],nums[start]);
         int pivot = nums[start];
         int i = start;
         for(int j = start+1;j<=end;j++){
                if(nums[j]<pivot){
                    i++;
                    swap(nums[i],nums[j]);
                }
            }
            swap(nums[start],nums[i]);
        return i;
        }
       
};