leetcode 215. Kth Largest Element in an Array最大k值

最新推荐文章于 2024-09-09 23:55:06 发布

原创最新推荐文章于 2024-09-09 23:55:06 发布 · 150 阅读

0 ·

CC 4.0 BY-SA版权

leetcode刷题专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一种使用快速排序定位法寻找未排序数组中第K大元素的方法。通过该算法，仅需O(n)的时间复杂度即可高效找到指定位置的元素。文章提供了具体的实现思路及代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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.

Example 1:
题目描述：找到第 k 大的元素。
Input: [3,2,1,5,6,4] and k = 2
Output: 5
Example 2:

Input: [3,2,3,1,2,4,5,5,6] and k = 4
Output: 4
思路：快速排序定位法因为要找k值所以只需排k的部分
时间复杂度o（n）
定位思路最左边定义为一个哨兵，然后比k大的放左边，比k小的放右边
如果碰到相反的则交换
如果左边地址坐标大于等于右面退出
然后交换哨兵与右边坐标位置直到找到k-1位置为止
代码：

class Solution {
public:
   
    int findKthLargest(vector<int>& nums, int k) {
        int left=0,right=nums.size()-1;
        while(left<right){
            int j=partion(nums,left,right);
            if(j==k-1)
                break;
            else if(j>k-1)
                right=j-1;
            else
                left=j+1;
        }
        return nums[k-1];
        
    }
    int partion(vector<int>& nums,int left,int right){
        int p=nums[left];
        int l=left+1;
        int r=right;
        while(true){
            while(nums[l]>=p&&l<right)l++;
            while(nums[r]<=p&&r>=l)r--;
            if(l>=r)break;
            swap(nums[l],nums[r]);
        }
        swap(nums[left],nums[r]);
        return r;
    }
};