Leetcode 215 数组中的第K个最大元素

最新推荐文章于 2024-08-17 15:37:21 发布

Coding_Bug_Lu

最新推荐文章于 2024-08-17 15:37:21 发布

阅读量167

点赞数

文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/weixin_43017004/article/details/126424994

版权

数据结构与算法专栏收录该内容

87 篇文章 0 订阅

订阅专栏

本文介绍了一种使用快速排序算法寻找给定整数数组中第k大的元素的方法。通过随机化和递归调用，代码展示了如何在C++中实现quicksort和randompartition函数，以在最坏情况下达到平均时间复杂度。

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

在这里插入图片描述

代码

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        srand(time(0));
        int q = quicksort(nums, 0, nums.size() - 1, nums.size() - k);
        return nums[q];
    }

    int quicksort(vector<int>& nums, int l, int r, int index) {
        int pos = randompartition(nums, l, r);
        if (pos == index) return pos;
        else {
            return pos < index ? quicksort(nums, pos + 1, r, index) : quicksort(nums, l, pos - 1, index);
        }
    }

    int randompartition(vector<int>& nums, int l, int r) {
        int i = rand() % (r - l + 1) + l;
        swap(nums[i], nums[r]);
        int pos = partition(nums, l, r);
        return pos;
    }

    int partition(vector<int>& nums, int l, int r) {
        int i = l - 1;
        for (int j = l; j <= r - 1; j++) {
            if (nums[j] <= nums[r]) {
                i++;
                swap(nums[i], nums[j]);
            }
        }
        swap(nums[i+1], nums[r]);
        return i + 1;
    }
};