您的位置:首页 > 其它

线性时间内从一个数组中找出第K个最小的元素

2013-04-29 10:54 309 查看
/*
思路:用调优的快速排序,首先随机选择一个pivot对数组进行划分,左边小于pivot,右边大于等于pivot,
所以我们计算左边小于pivot(加上pivot)的个数count总共有多少,
如果等于k,正是我们所要的,返回x[k];
如果大于k,说明第k小的数在左边,那就在左边进行我们的递归;
否则,在右边,那么说明右边的第k-count小的数就是我们所要的,在右边进行我们的递归。
这样的平均时间复杂度为O(N*logK),线性的、
*/
#include <stdio.h>
#include <stdlib.h>
/*交换x[i]与x[j]的值*/
void swap(int x[],int i, int j)
{	int t = x[i];
x[i] = x[j];
x[j] = t;
}
/*生成一个在有效范围内的随机数*/
int randint(int l, int u)
{	return l + (RAND_MAX*rand() + rand()) % (u-l+1);
}

/*递归查找第K个最小值*/
int GetMinK(int x[],int l,int u,int k)
{
int i, j;
int t, temp;
if (l >= u)
return 0;
swap(x,l, randint(l, u));
t = x[l];
i = l;
j = u+1;
/*在左边找个比t小的,在右边找个比t大的,然后交换,一直到两个标识位i,j相交*/
for (;;) {
do i++; while (i <= u && x[i] < t);
do j--; while (x[j] > t);
if (i > j)
break;
swap(x,i,j);
}
swap(x,l, j);
if(j==k)
return x[k];
else if (j < k)
GetMinK(x,j+1, u, k);
else if (j > k)
GetMinK(x,l, j-1, k);
}
int main()
{
int x[]={2,12,11,1,5,10,9,7,8,6};
int k=3;
printf("第%d小元素为:(从0开始)%d ",k,GetMinK(x,0,9,k));
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐