线性时间找第K小的数

用了快排的思想,区别是这里每次只是选择性地排某一边,而快排是两边都排,由于是快排,所以最坏情况不是O(n)

#include <stdio.h>
#include <stdlib.h>
int k,n,a[1000000];
void m_swap(int *a,int *b){ int t=*a;*a=*b;*b=t;}
int qsort(int *left,int *right)
{
	int *tl=left,*tr=right-1;
	int t=*left;
	while(tl<tr)
	{
		while(*tr>t) tr--;
		if(tl>=tr) break;
		m_swap(tr,tl);
		while(*tl<t) tl++;
		if(tl>=tr) break;
		m_swap(tr,tl);

	}
	t=tl-a;
	if(t==k) return *tl;
	else if(t>k) return qsort(left,tl);
	else return qsort(tl+1,right);
}
int main()
{
	scanf("%d %d",&n,&k);
	--k;
	for(int i=0;i<n;++i)
		scanf(" %d",a+i);
	printf("%d\n",qsort(a,a+n));
	return 0;
}


posted @ 2014-09-01 21:40  Netop  阅读(280)  评论(0)    收藏  举报