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

最新推荐文章于 2024-01-29 11:23:48 发布

1ts0k

最新推荐文章于 2024-01-29 11:23:48 发布

阅读量254

点赞数 1

分类专栏： # 排序文章标签：算法

本文链接：https://blog.csdn.net/wetqadsg/article/details/127892077

版权

排序专栏收录该内容

2 篇文章 0 订阅

订阅专栏

P1923 【深基9.例4】求第 k 小的数 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

思路 1：用sort 会tle O(nlog2n)

思路 2：用快速排序的分治思想 O(n)

代码如下：

#include <iostream>
#include <algorithm>
using namespace std;
int a[5000010];
int n, k;
void quick_sort(int l, int r)
{
	if (l >= r) return;
	int i = l - 1, j = r + 1, x = a[(l + r) >> 1];
	while (i < j)
	{
		do i++; while (a[i] < x);
		do j--; while (a[j] > x);
		if (i < j) swap(a[i], a[j]);
	}
	if (k <= j) quick_sort(l, j);
	else quick_sort(j + 1, r); //只sortk所在的区间 其余区间不用考虑不影响第k个小是什么 
}
int main()
{
	scanf("%d%d", &n, &k);
	for (int i = 0; i < n; i++) scanf("%d", &a[i]);
	
	quick_sort(0, n-1);
	
	printf("%d", a[k]);
	
	return 0;
}