LeetCode215-数组中的第K个最大元素-快速排序-快速选择-模板

最新推荐文章于 2022-09-06 20:51:56 发布

李烦烦搞快点

最新推荐文章于 2022-09-06 20:51:56 发布

阅读量134

点赞数

分类专栏： LeetCode&&笔试题文章标签： leetcode 算法快速排序快速选择模板题

本文链接：https://blog.csdn.net/Mr_Ghost812/article/details/124484713

版权

LeetCode&&笔试题专栏收录该内容

291 篇文章 0 订阅

订阅专栏

原题链接
在这里插入图片描述

Note：还是得背快拍的模板

在快排的基础上进行一些改进，快排每次劈成两半之后，看看要找的位置是在哪一边，就进一步去排哪一边就好

注意问的是第K大元素，我们让数组从大到小排序，那么nums[k - 1]就是结果了（题里的K是从1开始的）

代码如下：

class Solution {
public:
    int qs(vector<int>& nums, int l, int r, int k){
        if(l == r)  return nums[k];
        int x = nums[l], i = l - 1, j = r + 1;
        while(i < j){
            do i ++; while(x < nums[i]);
            do j --; while(x > nums[j]);
            if(i < j)   swap(nums[i], nums[j]);
        }
        if(k <= j) return qs(nums, l, j, k);
        else    return qs(nums, j + 1, r, k);
    }

    int findKthLargest(vector<int>& nums, int k) {
        return qs(nums, 0, nums.size() - 1, k - 1);
    }
};