码迷,mamicode.com
首页 > 编程语言 > 详细

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

时间:2020-02-22 13:39:48      阅读:58      评论:0      收藏:0      [点我收藏+]

标签:https   size   else   int   一个   array   技术   pre   就是   

题目

题意:找到一个数组里第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);
        
    }
};

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

标签:https   size   else   int   一个   array   技术   pre   就是   

原文地址:https://www.cnblogs.com/dacc123/p/12344804.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!