leetcode 215. Kth Largest Element in an Array

最新推荐文章于 2025-08-11 15:06:51 发布

Snail_Walker

最新推荐文章于 2025-08-11 15:06:51 发布

阅读量223

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode 文章标签： leetcode 215 kth largest

本文链接：https://blog.youkuaiyun.com/c602273091/article/details/54263196

Leetcode 专栏收录该内容

175 篇文章

订阅专栏

本文介绍了一个使用快速选择算法实现寻找第K大的元素的C++代码示例。通过定义一个Solution类，该类包含两个主要成员函数：partition和findKthLargest。partition函数用于将数组分为两部分，findKthLargest函数利用partition函数找到第K大的元素。

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

class Solution {
public:
    int partition(vector<int>& nums, int left, int right) {
        int pivot = nums[left];
        int l = left + 1, r = right;
        while (l <= r) {
            if (nums[l] < pivot && nums[r] > pivot)
                swap(nums[l++], nums[r--]);
            if (nums[l] >= pivot) l++; 
            if (nums[r] <= pivot) r--;
        }
        swap(nums[left], nums[r]);
        return r;
    }

    int findKthLargest(vector<int>& nums, int k) {
        int left = 0, right = nums.size() - 1;
        while (true) {
            int pos = partition(nums, left, right);
            if (pos == k - 1) return nums[pos];
            if (pos > k - 1) right = pos - 1;
            else left = pos + 1;
        }
    }
};