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

快速排序时间复杂度: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;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值