leecode第二百一十五题（数组中的第K个最大元素）

最新推荐文章于 2025-02-03 15:37:10 发布

weixin_30699741

最新推荐文章于 2025-02-03 15:37:10 发布

阅读量73

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/CJT-blog/p/10721869.html

本文介绍了一种使用快速排序算法寻找数组中第K大的元素的方法。通过修改快速排序算法，实现了一种高效的查找算法，该算法在平均情况下具有O(n)的时间复杂度。文章详细介绍了算法的实现过程，并提供了C++代码示例。

class Solution {
public:
    int quick_sort_version(vector<int>& nums, int k,int begin,int end,int len)
    {
        if(end-begin<1)
            return nums[begin];//这里有点小意外，本来写的是nums[0]，这是不对的，比如案例[1,2],k=1
        
        int res_beg=begin,res_end=end;//快排
        bool flag=true;
        while(res_beg!=res_end)
        {
            if(nums[res_beg]>nums[res_end])
            {
                int temp=nums[res_beg];
                nums[res_beg]=nums[res_end];
                nums[res_end]=temp;
                if(flag)
                    res_beg++;
                else
                    res_end--;
                flag=!flag;
            }
            else
            {
                if(flag)
                    res_end--;
                else
                    res_beg++;
            }
        }
        
        if(len-res_beg>k-1)//如果右面数比k-1多，就只在右面查找
            return quick_sort_version(nums,k,res_beg+1,end,len);
        else if(len-res_beg<k-1)//比k-1少，就只在桌面查找
            return quick_sort_version(nums,k,begin,res_beg,len);
        else//如果刚好相等，直接返回这个数值即可
            return nums[res_beg];
    }
    
    int findKthLargest(vector<int>& nums, int k) {
        int len=nums.size();
        return quick_sort_version(nums,k,0,len-1,len-1);//类似快排
    }
};