使用快速排序，求最大的k个数

最新推荐文章于 2022-04-01 21:28:36 发布

原创最新推荐文章于 2022-04-01 21:28:36 发布 · 770 阅读

0 ·

CC 4.0 BY-SA版权

编程专栏收录该内容

13 篇文章

订阅专栏

本文介绍了如何利用快速排序的思路，有效地找到一个数组中最大的k个数。通过分治策略，将问题分解，并在排序过程中找到前k大元素，避免了全量排序的时间消耗。

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

使用选择排序的思想：时间复杂度o(n*k)

#include<iostream>
using namespace std;
void selectMax(int arr[],int n,int k)
{
	for(int j = 0; j < k; ++j)
	{
		for(int i = 0; i < n - j - 1; ++i)
		{
			if(arr[i] > arr[i + 1])
				swap(arr[i],arr[i + 1]);
		}
	}
}
void selectSort(int arr[],int n)
{
	for(int i = 0; i < n - 1; ++i)
	{
		for(int j = 0; j < n - i - 1; ++j)
		{
			if(arr[j] > arr[j + 1])
				swap(arr[j],arr[j + 1]);
		}
	}
}
void swap(int &a,int &b)
{
	int temp = a;
	a = b;
	b = temp;
}
int main(void)
{
	int arr[] = {3, 5, 4, 8, 9, 10, 1, 2};
	//selectSort(arr,8);
	selectMax(arr,8,2);//最的两个数位于数组的最后。
	for(int i = 0; i < 8; ++i)
	{
		cout<<arr[i]<<endl;
	}

	system("pause");
	return 0;
}

快速排序的思想解决。o(n*logk)

#include<iostream>
using namespace std;
int partition(int arr[],int p,int r)//每一次划分所产生的结果
{
	int x = arr[r];//选择末尾元素作为枢轴
	int i = p -1;
	for(int j = p; j < r; ++j)
	{
		if(arr[j] > x)
		{
			i++;
			swap(arr[i],arr[j]);
		}
	}
	swap(arr[i + 1],arr[r]);
	return i + 1;
}

//begin代表开始的位置，end代表的结束的位置，k为最大的k+1个数
void kBig(int arr[],int begin,int end,int k)
{
	int i = partition(arr,begin,end);
	if(k == i)//如果划分的结果刚好为枢轴所在的位置，则返回
		return;
	else if(k > i)//如果枢轴的位置小于k,则递归的进入右区域
	{
		kBig(arr,i + 1,end,k);
	}
	else//否则进入左区域
	{
		kBig(arr,begin,i - 1,k);
	}
}

int main(void)
{
	int arr[] = {2, 8, 7, 1, 3, 5, 6, 4, 11, 13, 14, 15, 19, 16, 20, 21, 78, 32, 100, 10};
	partition(arr,0,7);
	kBig(arr,0,19,4);
	for(int i = 0; i < 20; ++i)
	{
		cout<<arr[i]<<endl;
	}
	system("pause");
	return 0;
}