215 数组中的第K个最大元素

hahahhahahahahah1

于 2023-09-25 23:02:05 发布

阅读量311

点赞数

CC 4.0 BY-SA版权

文章标签： java 算法数据结构

本文链接：https://blog.youkuaiyun.com/hongyuyahei/article/details/133281979

文章介绍了如何使用快速选择算法（quickSelect）实现时间复杂度为O(n)的查找第k大元素，通过随机选择基数并划分大于、等于和小于基准的元素，递归地缩小搜索范围。

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

在这里插入图片描述
满足时间复杂度o(n)的方法：
快排的思想

class Solution{
public:
	int findKthLargest(vector<int>& nums,int k){
		return quickSelect(nums,k);
	}
private:
	int quickSelect(vector<int>& nums,int k){
		//随机选择基数
		int privot=nums[rand()%nums.size()];
		//将大于、小于、等于privot的元素划分至big\small\equal中
		vector<int> big,equal,small;
		for(int num:nums){
			if(num>privot){
				big.push_back(num);
			}else if(num<privot){
				small.push_back(num);
			}else{
				equal.push_back(num);
			}
		}
		//第k大元素在big中，递归划分
		if(k<=big.size()){
			return quickSelect(big,k);
		}
		//第k大元素(第n+1-k小元素)在small中，递归划分
		if(nums.size()+1-k<=small.size()){
			//原先第k大，在small中k需减去big和equal集合的大小
			return quickselect(small,k-(nums.size()-small.size()));
		}
		//第k大元素在equal中，直接返回privot
		return privot;
		
	}
};