找到给定线性表中第K大元素

最新推荐文章于 2024-07-22 10:30:49 发布

TonyLand001

最新推荐文章于 2024-07-22 10:30:49 发布

阅读量1.1k

点赞数

CC 4.0 BY-SA版权

分类专栏：笔试面试 C++

本文链接：https://blog.youkuaiyun.com/richardzjut/article/details/39999039

C++ 同时被 2 个专栏收录

6 篇文章

订阅专栏

笔试面试

1 篇文章

订阅专栏

本文介绍了一种改进的快速排序算法，用于高效地找出数组中第K大的元素。该方法仅对包含目标元素的子数组进行排序，从而显著降低了时间复杂度，使其接近O(n)。

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

思路：改进快排算法。假设需要找到第K大元素，则只针对K序号所在的分支进行快速排序。

时间复杂度：接近O(n)

实现：

#include <iostream>
using namespace std;

int partition(int arr[], int left, int right)
{
	int pivot = arr[left];
	if(left<right)
	{
		while(left<right)
		{
			while(left<right&&arr[right]>=pivot)
				--right;
			arr[left]=arr[right];
			while(left<right&&arr[left]<=pivot)
				++left;
			arr[right]=arr[left];
		}
	}
	arr[right] = pivot;
	return right;
}

int findKthMax(int arr[], int left, int right, int k_index)
{
	int curr = partition(arr, left, right);
	if(curr<k_index-1)
		return findKthMax(arr,curr+1,right,k_index);
	else if(curr>k_index-1)
		return findKthMax(arr,left,curr-1,k_index);
	
	return curr;
}

int main()
{
	int arr[] = {3,2,5,1,9,7,6,10};
	int size = sizeof(arr)/sizeof(int);
	int resultIndex = findKthMax(arr,0,size-1,8);
	cout<<"the N-th max element is:"<<arr[resultIndex]<<endl;

	return 0;
}