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:
- 思路分析:最简单的做法就是将给定的无序数组排好序,输出排序后的数组中第k大的元素。考虑到算法的时间复杂度,我们可以选择快速排序,这样子能够将复杂度降低到O(N*logN),而快排可以用STL中的排序函数sort,默认对数组从小到大进行排序,sort函数是qsort函数的改进版。
- 代码实现:
#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:
- 思路分析:由于题目要求的只是数组中第k大的元素,所以我们不必将数组排序,而可以学习快排的思想:先声明三个数组num1,num2,num3;然后选定数组中的某个元素作为中心点,遍历数组的每个元素,如果小于中心点就放在数组num1中,等于中心点就放在数组num2中,大于中心点就放在数组num3中;再将k值与num1,num2,num3进行比较,判断出下一步是递归findKthLargest函数还是输出最终值。
- 代码实现:
#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;
};