注:本文不对快速排序作任何解释,建议在对快速排序有一定了解后再阅览
一、问题分析
最简单的做法应该是直接选择先将集合排序(比如快速排序),然后直接以k为下标从有序集合中获取。但是这样做时间复杂度其实是比较大的。如果要想要提升一下效率,可以考虑在快速排序的原理下稍微做点修改。
二、修改快排
1、主元素的位置特殊性
在快速排序中,第一步是选取主元素(这里记为x),然后将小于主元素x的数放在x左边,剩下的所有大于x的数放在x右边。记x的下标为x_index,这里不难发现,此时x正是集合中第x_index(如果下标从0开始算,则是x_index+1)大的元素。
2、题目情景特殊性----只求一个数
按照快速排序的步骤,接下来是将左、右两个子集合分别重复上述的“选取主元素,其余元素按照大小放主元素左右两边”的操作,但事实题目只是要求找到一个第k大的元素。那么,左右两个子集是不是可以考虑舍去一个?(类似于二分法,只取一半)
显然可以!我们可以将k与x_index进行比较,分三种情况,当
- k = x_index 时,说明已经找到了
- k < x_index 时,说明第k大元素在x左边的子集里,可以舍去右边部分
- k > x_index 时,说明第k大元素在x右边的子集里,可以舍去左边部分
3、C代码实现:
#include <stdio.h>
int exchange(int *a, int i, int j){
int temp = a[i];
a[i] = a[j];
a[j] = temp;
}
int partition(int *a, int p, int r){
int x