题目五:
在数组中找到第k大的元素
你可以交换数组中的元素的位置
正确代码:
class Solution {
/*
* @param k : description of k
* @param nums : array of nums
* @return: description of return
*/
public :
int kthLargestElement(int n,vector<int> &nums) {
// write your code here
return nums[pat(nums,0,nums.size()-1,n-1)];
}
int pat(vector<int> &nums,int start,int end,int k){
int index=qsort(nums, start, end);
if(index==k){
return index;
}else if(index>k){
return pat(nums,start,index-1,k);
}else{
return pat(nums,index+1,end,k);
}
}
int qsort(vector<int> &nums,int start,int end){
int head=nums[start];
while(start<end){
while(start<end){
if(nums[end]>head){
nums[start]=nums[end];
start++;
break;
}
end--;
}
while(start<end){
if(nums[start]<head){
nums[end]=nums[start];
end--;
break;
}
start++;
}
}
nums[start]=head;
return start;
}
};
在度娘的帮助下很艰难的理解了这个算法。
主要是采用快速排序的思路。这里是从大到小排序,比轴值大,放在左侧,比轴值小,放在右侧,最后start=end的位置就是轴值的位置,返回轴值的位置,如果轴值的位置恰好等于k,说明轴值就是第k大,小于k,说明在右侧,大于k说明在左侧,递归调用。
我之前也写了我自己的算法,基本都是在77%上卡住了主要是时间超时,我发现我之前的代码有很多交换同一位置上的元素的操作,在数据很庞大的时候这些看似不起眼的运行时间累加起来也是很耗时的。