LeetCode之Kth Largest Element in an Array

最新推荐文章于 2024-11-19 02:25:22 发布

原创最新推荐文章于 2024-11-19 02:25:22 发布 · 472 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Leetcode

Leetcode 专栏收录该内容

168 篇文章

订阅专栏

本文介绍了一种利用快速排序的思想来寻找数组中第K大的元素的方法。通过不断分区，直到找到目标元素的位置。这种方法效率高，适用于大数据量的情况。

/*分治法，采用快排中用到的partition函数。*/
class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int left = 0, right = nums.size()-1;
        while(left <= right){
            int res = findKthLargest(nums, left, right);
            if(res == nums.size()-k) return nums[res];
            else if(res < nums.size()-k) left = res+1;
            else right = res-1;
        }
    }
    
    int findKthLargest(vector<int> &nums, int left, int right){
        int x = nums[right];
        int i = left-1;
        for(int j = left; j < right; ++j){
            if(nums[j] <= x){
                ++i;
                swap(nums[i], nums[j]);
            }
        }
        swap(nums[i+1], nums[right]);
        return i+1;
    }
};