Kth Largest Element in an Array

Question:

Find the kth largest element in an unsorted array. Note that it is the kth largest element in the sorted order, not the kth distinct element.
For example: Given [3,2,1,5,6,4] and k = 2, return 5.
Note: You may assume k is always valid, 1 ≤ k ≤ array’s length.

Solution1:

  1. 思路分析:最简单的做法就是将给定的无序数组排好序,输出排序后的数组中第k大的元素。考虑到算法的时间复杂度,我们可以选择快速排序,这样子能够将复杂度降低到O(N*logN),而快排可以用STL中的排序函数sort,默认对数组从小到大进行排序,sort函数是qsort函数的改进版。
  2. 代码实现:
#include <algorithm>
#include <vector.h>
class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        sort(nums.begin(), nums.end());     //默认为升序排序
        return nums[nums.size() - k];
    }
};

Solution2:

  1. 思路分析:由于题目要求的只是数组中第k大的元素,所以我们不必将数组排序,而可以学习快排的思想:先声明三个数组num1,num2,num3;然后选定数组中的某个元素作为中心点,遍历数组的每个元素,如果小于中心点就放在数组num1中,等于中心点就放在数组num2中,大于中心点就放在数组num3中;再将k值与num1,num2,num3进行比较,判断出下一步是递归findKthLargest函数还是输出最终值。
  2. 代码实现:
#include <vector.h>
class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int flag = nums[0];
        split(nums, nums.size(), flag);
        //kth element在大于flag的数组
        if (k <= nums3.size())
            return findKthLargest(nums3, k);
        //kth element在等于flag的数组
        else if (k > nums3.size() && k <= nums3.size() + nums2.size())
            return flag;
        //kth element在小于flag的数组
        else
            return findKthLargest(nums1, k - nums3.size() - nums2.size());
    }
    void split(vector<int>& nums, int size, int flag) {
        //注意clear!!!
        nums1.clear();  nums2.clear();  nums3.clear();
        for (int i = 0; i < size; i++) {
            if (nums[i] < flag) {
                nums1.push_back(nums[i]);
            }
            else if (nums[i] == flag) {
                nums2.push_back(nums[i]);
            }
            else {
                nums3.push_back(nums[i]);
            }
        }
    }
private:
    vector<int> nums1, nums2, nums3;
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值