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

数组中的第K个最大元素

时间:2018-07-30 23:03:03      阅读:206      评论:0      收藏:0      [点我收藏+]

标签:利用   快排思想   ++   dex   时间复杂度   color   cto   find   思想   

利用快排思想:先确定一个轴点,然后进行交换使得轴点左边的值比轴点小,轴点右边的值比轴点大,当我们找到轴点位于第N-K个位置的时候说明找到第K大的值。也可以反过来逆序。时间复杂度为O(N).

代码如下:

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int index=getKthindex(nums,k,0,nums.size()-1);
        return nums[index];
    }
    
    int getKthindex(vector<int>&a,int k,int lo,int hi){
        int mi=getPartion(a,lo,hi);
        if(mi==a.size()-k)return mi;
        else if(mi<a.size()-k)return getKthindex(a,k,mi+1,hi);
        else{
            return getKthindex(a,k,lo,mi-1);
        }
    }
    
    int getPartion(vector<int>&a,int lo,int hi){
        int pivot=a[lo];
        while(lo<hi){
            while(lo<hi&&pivot<=a[hi])hi--;
            swap(a[lo],a[hi]);
            while(lo<hi&&pivot>=a[lo])lo++;
            swap(a[lo],a[hi]);
        }
        return lo;
    }
};

 

数组中的第K个最大元素

标签:利用   快排思想   ++   dex   时间复杂度   color   cto   find   思想   

原文地址:https://www.cnblogs.com/inception6-lxc/p/9393268.html

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