如何在o(n)内查找第k小的元素

利用快排思想在o(n)内查找第k小的元素

题目概要：

 o(n)时间复杂度内求无序数组中的第k小元素。比如，4，2，5，11，6这样一组数据，第3小的元素就是5。

解题思想：

利用快排的思想（从小到大排列）我们可以将数组a[0...r]原地分解成a[0....p-1],a[p],a[p+1...r]三部分，

如果p+1=k,那么a[p]就是我们要找的元素；

如果p+1<k,那么说明第k小的元素出现在a[p+1...r]区间；

如果p+1>k,那么说明第k小的元素出现在a[0....p-1]区间；

（后两种情况我们在按照上面的思路递归在相应的区间查找即可）

代码:

#include<stdio.h>
#include<stdlib.h>

void swap(int* p1, int* p2)  //用于交换元素
{
	int temp = *p1;
	*p1 = *p2;
	*p2 = temp;
}
int select(int a[],int start,int end,int k) 
{
	// 利用快排，找出分区点，分解区间
	int pivot = a[end];
	int i = start;
	for (int j = start; j < end; j++)
	{
		if (a[j] < pivot) {
			swap(&a[i], &a[j]);
			i++;
		}
	}
	swap(&a[i], &a[end]);  
	// 进行情况判断
	if (i + 1 == k) return a[i];
	if (i + 1 < k)
		return select(a, i + 1, end, k);
	if (i + 1 > k)
		return select(a, 0, i - 1, k);
}

int main()
{
	int n,k;
	int a[20];
	scanf_s("%d%d", &n,&k);
	for (int i = 0; i < n; i++)
	{
		scanf_s("%d", &a[i]);
	}
	if (k<1 || k>n)  
		printf("error");
	else
	 printf("%d",select(a, 0, n - 1, k));

}