期望为线性时间的选择算法

Posted 2020-11-24 kirosola

该算法以快排为基础

int random_select(int A[], int p, int r, int i)
{
	if (p == r)
		return A[p];
	int q = random_partition(A, p, r);
	int k = q - p + 1;
	if (i == k)
		return A[q];
	else if (i < k)
		return random_select(A, p, q - 1, i);
	else
		return random_select(A, q + 1, r, i - k);
}

第三行：将数组进行划分。第四行：A[p...q]中元素个数为k。

检查A[q]是否为第i小的元素，是则返回。

如果i<k，要找的元素位于划分的低区。如果i>k，要找的元素位于划分的高区。

int random_partition(int A[], int p, int r)
{
	srand(time(NULL));
	int d = r - p + 1;
	int q = p + (rand() % d);//产生p到r的随机数
	swap(A[q], A[r]);
	return partition(A, p, r);
}
int partition(int A[], int p, int r)
{
	int x = A[r];
	int i = p - 1;
	for (int j = p; j <= r - 1; j++)
	{
		if (A[j] < x)
		{
			i++;
			swap(A[i], A[j]);
		}
	}
	swap(A[i + 1], A[r]);
	return i + 1;
}