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

最新推荐文章于 2024-09-14 13:16:14 发布

Lookatm1

最新推荐文章于 2024-09-14 13:16:14 发布

阅读量142

点赞数

文章标签： c++ leetcode

本文链接：https://blog.csdn.net/Lookatm1/article/details/116209493

版权

题目描述：

在未排序的数组中找到第 k 个最大的元素。请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

示例 1:

输入: [3,2,1,5,6,4] 和 k = 2
输出: 5

链接：https://leetcode-cn.com/problems/kth-largest-element-in-an-array

解题思路：

一开始想用sort，面试中肯定不能投机取巧，所以考虑先手动排序，再输出，这样做会超时，只要把数组全排列就会超时。于是转为分治，选原数组的第一个数为基准数，比他大的放在一边，小的放在另一边，如果大数组的size=k-1，那么基准数就是要找的第k大的数，输出基准数。如果size>=k，说明要找的数在大数组中，对大数组进行搜索；如果size<k，说明要找的数在小数组中，对小数组进行分治搜索，注意这里要把k减去大数组的大小，再减1（减掉基准数）。

c++代码：

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        vector<int> numsmin;
        vector<int> numsmax;
        while(!nums.empty()){
            int i=nums[0];
            for(int j=1;j<nums.size();j++){
                if(nums[j]<=i) numsmin.push_back(nums[j]);
                else numsmax.push_back(nums[j]);
            }
            if(k==numsmax.size()+1) return i;
            else if(k<=numsmax.size()){
                nums=numsmax;
                numsmax.clear();
                numsmin.clear();
            }
            else{
                k-=(numsmax.size()+1);
                nums=numsmin;
                numsmax.clear();
                numsmin.clear();
            }
        }
        return 0;
    }
};