算法分析与设计——2.3线性时间选择

问题描述：给定线性序集中n个元素和一个整数k，1<=k<=n，要求找出这n个元素中第k小的元素。

例如：a[]={0,2,1,3,4,5,8}，k=1时，要找的元素就是最小元素a[0]=0。

算法思想：利用随机划分算法。时间复杂度T(n)=O(n)。

template<typename T>
int Partition_new(T a[], int p, int r)
{
	T x = a[r];
	int i = p - 1;
	for (int j = p; j <= r; j++)
	{
		if (a[j] <= x)
		{
			i++;
			if (i != j)
				swap(a[i], a[j]);
		}
	}
	return i;
}

template<typename T>
int RandomizedPartition(T a[], int p, int r)
{
	int i = rand() % (r - p + 1) + p;
	swap(a[i], a[p]);
	return Partition_new(a, p, r);
}

template<typename T>
T RandomizedSelect(T a[], int p, int r, int k)
{
	if (p == r)return a[p];
	int i = RandomizedPartition(a, p, r);
	int j = i - p + 1;
	if (k <= j)return RandomizedSelect(a, p, i, k);
	else return RandomizedSelect(a, i + 1, r, k - j);
}