LeetCode 215. Kth Largest Element in an Array(排序)
题意:找到一个数组里第K大的数字。
题解:我们当然可以排序好了,之后,选择第K大的数字。但是这样做一点技术含量也没有。
排序算法选用快排。寻找第K大的数字,不必把数组完全排完序之后,再找第K大。快排中是选取一个数字,把大于它的放在右边,小于它的放在左边,在递归的时候,我们判断k 和右边数字的个数,如果k大,说明第k大的数字在左边,于是我们去找左边的部分,就不用递归排序右边的部分了,如果k小。
快排的平均效率是O(nlog(n)) 而,这道题目用快排的效率就是O(n)的效率,实际上是O((1-1/2^n)n) 当n趋近于无穷大的时候,就是O(n)。
class Solution {
public:
int findKthLargest(vector<int>& nums, int k) {
return fun(nums,0,nums.size()-1,k);
}
int fun(vector<int>& nums,int l,int r,int k)
{
int s = l+1;
int e = r;
while(s<=e)
{
while(e>=s&&nums[e]>nums[l])
{
e--;
}
while(s<=e&&nums[s]<nums[l])
{
s++;
}
if(s<=e)
{
swap(nums[e],nums[s]);
e--;
s++;
}
}
swap(nums[l],nums[e]);
int x = r - e;
if(x==k-1)
return nums[e];
else if(x>k-1)
return fun(nums,e+1,r,k);
else
return fun(nums,l,e-1,k-x-1);
}
};