原题目:
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.
题目大意如下:
给你一个未排序的数组找到数组当中第K大的那个数,注意是返回有序数组当中第K大的那个数,并不是第K远的那个数。
解题思路:
这道题题目意思当然是很简单的,做法也有很多种,但我们需要做的就是要找到时间复杂度最小的那一种方法(最后可以知道最小复杂度为稳定的O(n)),在这里我顺便也把其他方法简单列举一下。
1.快速排序(利用sort函数,复杂度为O(nlogn)。)
class Solution {
public:
int findKthLargest(vector<int>& nums, int k) {
sort(nums.begin() , nums.end()) ;
return nums[nums.size() - k] ;
}
};
2.利用快速排序的选择思想(最坏情况O(n²),最好情况O(n))
class Solution {
public:
int findKthLargest(vector<int>& nums, int k) {
int left = 0, right = nums.size() - 1;
while (1) {
int pos = partition(nums, left, right);
//如果选择的pivot符合条件就返回
if (pos == k - 1) return nums[pos];
//选择的pivot太小了,我们取大的那部分继续
if (pos > k - 1) right = pos - 1;
//选择的pivot太大了,取小的那部分继续
else left = pos + 1;
}
}
//这个化分区间的方法可以数组划分成三个区间,左边的不小于大pivot,右边的不大于pivot
int partition(vector<int>& v , int left ,int right){
int pivot = v[left];
int l = left + 1, r = right;
while (l <= r) {
if (v[l] < pivot && v[r] > pivot)
swap(v[l++], v[r--]);
if(v[l] >= pivot) l++;
if(v[r] <= pivot) r--;
}
swap(v[left], v[r]);
return r;
}
};
3.利用随机选取pivot优化(保证时间复杂度为O(n))
int partition(vector<int>& v , int left ,int right){
//进行随机选取pivot
srand((unsigned)time(NULL)) ;
int random = rand()%(right - left + 1) + left ;
swap(v[left] , v[random]) ;
int pivot = v[left];
int l = left + 1, r = right;
while (l <= r) {
if (v[l] < pivot && v[r] > pivot)
swap(v[l++], v[r--]);
if(v[l] >= pivot) l++;
if(v[r] <= pivot) r--;
}
swap(v[left], v[r]);
return r;
}
};
运行结果对比:
第一种:
第二种:
第三种: