数组中的第K个最大元素

Posted 追逐面包和牛奶

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了数组中的第K个最大元素相关的知识,希望对你有一定的参考价值。

利用快排思想:先确定一个轴点,然后进行交换使得轴点左边的值比轴点小,轴点右边的值比轴点大,当我们找到轴点位于第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个最大元素的主要内容,如果未能解决你的问题,请参考以下文章

数组中的第K个最大元素 优先队列

LeetCode.215-数组中的第K个最大元素

Leetcode 215. 数组中的第K个最大元素 By Python

215. 数组中的第K个最大元素

#yyds干货盘点# LeetCode 腾讯精选练习 50 题:数组中的第K个最大元素

LeetCode:数组中的第K个最大元素215