DSOJ Kth Number（第k大的数）

题目链接

#include<stdio.h>		//查找数组中第k小的元素
#define MAX 100
/*
算法思想：
模仿快排的做法，首先去数组中任何一个元素（不妨取第一个），将数组分为两部分，s[0,..p-1],和s[p+1,...n-1],
其中左半部分的值都不大于s[p]，右半部分的值都不小于s[p]

此时如果p==k-1,则说明s[p]即为所要找的第k小的元素，返回s[p]
否则，如果p>k-1，说明第k小的元素一定在s[0,...p-1]中则再在s[0,....,p-1]中查找
否则，说明第k小的元素在s[p+1,...,n-1]中，而且是s[p+1,...,n-1]中第k-p-1小的元素

综上，利用递归即可实现
*/

int FindKth(int *Array, int n, int k)
{
	int left, right, pivot;
	left = 0;
	right = n - 1;
	pivot = Array[0];				//选取第一个数组元素作为轴值
	while (left < right)
	{
		while (left < right&&Array[right] >= pivot)
			right--;
		Array[left] = Array[right];
		while (left < right&&Array[left] <= pivot)
			left++;
		Array[right] = Array[left];
	}
	Array[left] = pivot;			//left即为轴值所在的位置
	if (left == k - 1)
		return Array[left];
	else if (left>k - 1)
		return FindKth(Array, left, k);
	else
		return FindKth(Array + left + 1, n - left - 1, k - left - 1);
}

int main()
{
	int Array[MAX];
	int k, n, i;
	scanf("%d", &n);
	for (i = 0; i < n; i++)
		scanf("%d", Array + i);
	scanf("%d", &k);
	printf("%d\n", FindKth(Array, n, k));
	return 0;
}