洛谷P1923 【深基9.例4】求第 k 小的数进阶解法

最新推荐文章于 2024-10-14 20:29:56 发布

原创最新推荐文章于 2024-10-14 20:29:56 发布 · 249 阅读

1 ·

CC 4.0 BY-SA版权

程序设计代码共享专栏收录该内容

183 篇文章

订阅专栏

该博客详细介绍了如何实现一个快速查找算法，并在C++中给出了一段代码示例。这个算法用于在已排序的数组中查找指定位置的元素，通过分治策略达到高效查找的目的。在主函数中，首先读取数组大小n和目标位置k，然后对数组进行快速查找并输出结果。

#include<cstdio>

int n, k;
long long a[5000001];

long long quickfind(int low, int high, int number)
{
	if (low == high)
		return a[low];
	int i = low, j = high + 1;
	long long key = a[low], temp;
	for (;;)
	{
		for (; a[++i] < key;)
			if (i == high)
				break;
		for (; a[--j] > key;)
			if (j == low)
				break;
		if (i >= j)
			break;
		temp = a[i];
		a[i] = a[j];
		a[j] = temp;
	}
	temp = a[j];
	a[j] = a[low];
	a[low] = temp;
	if (j == k)
		return a[j];
	else if (k > j)
		return quickfind(j + 1, high, k - j);
	else if (k < j)
		return quickfind(low, j - 1, k);
}

int main()
{
	scanf("%d%d", &n, &k);
	for (int i = 0; i < n; i++)
		scanf("%lld", &a[i]);
	printf("%lld", quickfind(0, n - 1, k));
	return 0;
}