在LeetCode上看到一个题目,第一眼看的感觉就是排序解决,然后思考了下堆排序不太合适,而快排速度很快,然后在快速排序的基础上进行了优化,写下来记录一下。
题目
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.
题目分析
- 并不需要全部排序
- 注意小细节,如数据量为1的时候
代码(非递归)
public static int findKthLargest(int[] nums, int k) {
//把大的放左边,小的放右边,分割成两部分。当k小于中间数,往左查找,大于则往右查找,灵活运用快排
int left=0;
int mid=nums.length;
int right=nums.length-1; //使用最右边的作为划分点
while(true){
int i=left-1;
int j=right;
mid=nums[right]; //记录划分点
while(i!=j){
while(nums[++i]>mid&&i<j) //左边找到小于标记点的
if(i==right) break; //一直没找到
while(nums[--j]<mid&&i<j)//右边找到大于标记点的
if(j==left) break;
if(i>=j) break; //如果此时i>j跳出
//交换两个找到的标记点
int temp=nums[i];
nums[i]=nums[j];
nums[j]=temp;
}
//将划分点交换
int temp=nums[i];
nums[i]=nums[right];
nums[right]=temp;
//i+1是因为i是从0开始的,当找到的划分点的位置小于我要的k位置,那么就以标记点为最左边元素,在他的右边继续找,而左边的可以不管了,与我们要找的无关,相反如果要找的划分点大于k往右找,而如果等于,那么这个元素就是我们要找的了,退出循环结束
if(i+1<k){
left=i+1;
}else if(i+1>k){
right=i-1;
}else{
break;
}
if(left>=right) break;
}
return nums[k-1];
}
递归的方式就不贴了,比较简单。