LeetCode 215. Kth Largest Element in an Array(排序)

Posted 2021-03-08 Shendu.cc

题目

题意：找到一个数组里第K大的数字。

题解：我们当然可以排序好了，之后，选择第K大的数字。但是这样做一点技术含量也没有。

       排序算法选用快排。寻找第K大的数字，不必把数组完全排完序之后，再找第K大。快排中是选取一个数字，把大于它的放在右边，小于它的放在左边，在递归的时候，我们判断k 和右边数字的个数，如果k大，说明第k大的数字在左边，于是我们去找左边的部分，就不用递归排序右边的部分了，如果k小。

快排的平均效率是O(nlog(n)) 而，这道题目用快排的效率就是O(n)的效率，实际上是O((1-1/2^n)n) 当n趋近于无穷大的时候，就是O(n)。

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
      
          return fun(nums,0,nums.size()-1,k);
        
    }
    
    int fun(vector<int>& nums,int l,int r,int k)
    {   
        int s = l+1;
        int e = r;
        
        while(s<=e)
        {
            while(e>=s&&nums[e]>nums[l])
            {
                e--;
            }
            
            while(s<=e&&nums[s]<nums[l])
            {
                s++;
            }
            if(s<=e)
            {
                swap(nums[e],nums[s]);
                e--;
                s++;
            }
            
            
        }
        
        swap(nums[l],nums[e]);
        
        int x = r - e;
        if(x==k-1)
            return nums[e];
        else if(x>k-1)
            return fun(nums,e+1,r,k);
        else 
            return fun(nums,l,e-1,k-x-1);
        
    }
};