寻找第K大（借用快排思想）

最新推荐文章于 2022-08-19 09:29:46 发布

同志仍需努力！

最新推荐文章于 2022-08-19 09:29:46 发布

阅读量190

点赞数

CC 4.0 BY-SA版权

分类专栏：算法设计与分析 C/C++ 文章标签：寻找第k大 TOP k 快排

本文链接：https://blog.csdn.net/t18438605018/article/details/119870473

本文介绍了如何运用快速排序的思想解决寻找数组中第K大元素的问题。通过一次快排，确定基准位置，并根据基准位置与(n-K)的关系判断继续在左或右子数组中查找，避免了全数组排序，将时间复杂度优化至O(n)。讨论了递归实现可能遇到的栈溢出问题，并鼓励读者提出改进意见。

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

1.题目：牛客网 NC88 （寻找第K大）

描述
有一个整数数组，请你根据快速排序的思路，找出数组中第K大的数。
给定一个整数数组a,同时给定它的大小n和要找的K(1<K<n)，请返回第k大的数(包括重复的元素，不用去重)，保证答案存在。

要求时间复杂度O(n)

示例1
输入：
[1,3,5,2,2],5,3
返回值：
2

2.解题思路
这题给了提示，可以借用快排的思想。同时要注意对事件复杂度的要求是O(n)。
对于快排不熟悉的，建议看下我的这篇博文：快速排序详细讲解和代码实现
先给大家介绍第一种思路：先对A进行快速排序（从小到大），然后返回第K大的数

class Solution {
   
   
public:
    int findKth(vector<int> a, int n, int K) {
   
   
        // write code here
        quickSort(a,0,n-1);
        return a[n-K];
    }
    void quickSort(vector<int> &A,int low,int high)
    {
   
   
        if(low<high)
        {
   
   
            int pivotpos=Partition(A,low,high);
            quickSort(A,low,pivotpos-1);
            quickSort(A,pivotpos+1,high);
        }
    }
    int Partition(vector<