您的位置:首页 > 其它

寻找第K小元素O(N)算法

2012-03-16 21:22 190 查看
快速排序时间复杂度:O(n*lgn)

但是没必要完整排序后再O(1)选出第k元素。

思想:利用快排分区思想,分界点为q,左边的数小于等于q,右边的数大于q这个规则。如果k小于左边区间个数则以左区间递归,否则在右区间递归。
#include <stdio.h>
#define MAX 100000
void swap(int *a, int *b){/*交换*/
int temp;
temp = *a;
*a = *b;
*b = temp;
}
int quicksort(int a[], int s, int e){
int i;
int l = s, r = e;
for(i = s; i< e;i++){ /*所有比a[e]小的元素都排前面*/
if(a[i] < a[e])
swap(&a[l++], &a[i]);
}
swap(&a[l], &a[e]);/*所有比a[e]大的排其后面*/
return l; /*找到区间中大于和小于最末元素的分界*/
}
int selectMinK(int a[], int s, int e, int k){
if(s == e)
return a[s];
int q = quicksort(a, s, e);
int num = q-s+1;/*计算左边区间元素个数*/
if(num == k)
return a[q];
if(num > k)
return selectMinK(a, s, q-1, k);/*递归搜索左边区间*/
else
return selectMinK(a, q+1, e, k-num);/*递归搜索右区间*/
}
int main(){
int a[MAX], n, k, ret,  i;
scanf("%d",&n);
for(i =0 ;i < n; i++){
scanf("%d",&a[i]);
}
scanf("%d",&k);
ret = selectMinK(a, 0, n-1 ,k);
printf("%d\n", ret);
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: