Kth Largest Element in an Array

来源:互联网 发布:mac如何卸载app 编辑:程序博客网 时间:2024/06/07 23:39

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;};
原创粉丝点击