LintCode(5)-第k大元素

最新推荐文章于 2024-10-02 23:40:03 发布

Msycamore

最新推荐文章于 2024-10-02 23:40:03 发布

阅读量99

点赞数

文章标签：算法

本文链接：https://blog.csdn.net/Qmingwt/article/details/125921055

版权

这篇博客介绍了如何通过快速排序算法在给定的整数数组中找到第k个最大元素。代码实现了一个名为`Solution`的类，包含`kthLargestElement`和`partition`两个方法。`partition`方法用于对数组进行划分，`kthLargestElement`方法则利用`partition`找出第k大的元素。整个过程通过递归完成。

摘要由CSDN通过智能技术生成

在数组中找到第 k 大的元素。ps:使用快排

class Solution {
public:
    /**
     * @param k: An integer
     * @param nums: An array
     * @return: the Kth largest element
     */
    int kthLargestElement(int k, vector<int> &nums) {
        // write your code here
        int n = nums.size();
        return partition(nums, 0, n-1, k);
    }
    int partition(vector<int>& nums, int start, int end, int k) {
        int left = start, right = end;
        int pivot = nums[left];
        
        while (left <= right) {
            while (left <= right and nums[left] > pivot) {
                left++;
            }
            while (left <= right and nums[right] < pivot) {
                right--;
            }
            if (left <= right) {
                swap(nums[left], nums[right]);
                left++;
                right--;
            }
        }
        if ((k-1) <= right) {
            return partition(nums, start, right, k);
        }
        if ((k-1) >= left) {
            return partition(nums, left, end, k);
        }
        return nums[k-1];
    }
};