利用快排思想找数组中第K小的数

最新推荐文章于 2021-07-16 13:28:24 发布

原创最新推荐文章于 2021-07-16 13:28:24 发布 · 271 阅读

1 ·

CC 4.0 BY-SA版权

本文介绍了一种利用快速选择算法寻找数组中第K个元素的方法，并探讨了如何进一步找出数组中小于该元素的所有数值。通过示例代码展示了算法的具体实现过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<iostream>
#include<vector>
using namespace std;
int QP(vector<int>arr, int begin, int end,int k)
{
	if (begin == end)return arr[begin];
	int first = begin;
	int last = end;
	int key = arr[first];
	while (first < last)
	{
		while (first < last&&key <= arr[last])
			last--;
		arr[first] = arr[last];
		while (first < last&&key >= arr[first])
			first++;
		arr[last] = arr[first];
	}
	arr[first] = key;
	if (first+1 == k)return arr[first];
	else if (first+1 < k)return QP(arr, first + 1, end, k);
	else return QP(arr, begin, first - 1, k);
}
int findk(vector<int>arr,int k)
{
	int len = arr.size();
	return QP(arr, 0, len - 1, k);
}
int main()
{
	vector<int>a = { 6,5,7,8,1,3,4,9,2,0 };
	int No = findk(a, 9);
	cout << No;
}

我在想数组从小到大前K个数是不是可以先把第K个数找出来，然后再遍历一遍数组把所有比该数字小的数字全部输出出来？