码迷,mamicode.com
首页 > 其他好文 > 详细

刷题215. Kth Largest Element in an Array

时间:2020-03-29 10:58:34      阅读:60      评论:0      收藏:0      [点我收藏+]

标签:元素   swap   art   line   int   小根堆   cout   auto   iss   

一、题目说明

题目215. Kth Largest Element in an Array,在一个无序数组中找第k大的元素。难度是Medium!

二、我的解答

这个题目最直观的解答是,先对数组排序,然后直接返回:

class Solution{
	public:
		int findKthLargest(vector<int>& nums,int k){
			sort(nums.begin(),nums.end());
			
			return nums[nums.size()-k];
		}
};

性能如下:

Runtime: 8 ms, faster than 97.67% of C++ online submissions for Kth Largest Element in an Array.
Memory Usage: 9.2 MB, less than 93.94% of C++ online submissions for Kth Largest Element in an Array.

三、优化措施

用小根堆实现,无需多言:

class Solution{
	public:
		//queue first in first out with priority delete
		int findKthLargest(vector<int>& nums,int k){
			//升序队列 ,小根堆 
			priority_queue<int,vector<int>,greater<int> > q;
			
			for(auto it: nums){
				q.push(it);
				if(q.size()>k){
					cout<<q.top()<<"->"; 
					q.pop();
				}
			}
			
			return q.top();
		}
};
Runtime: 12 ms, faster than 80.01% of C++ online submissions for Kth Largest Element in an Array.
Memory Usage: 9.5 MB, less than 39.39% of C++ online submissions for Kth Largest Element in an Array.

上面2个方法都不是最好的办法:方法1胜在简单易实现,方法2直观,方法3利用快速排序的思想:

class Solution{
	public:
		//利用快速排序的思想,不断将集合划分为左右两部分,
		//如果划分的位置pivot>k-1,则第k大的数在左边 
		//如果划分的位置pivot<k-1,则第k大的数在右边 
		int findKthLargest(vector<int>& nums,int k){
			int low = 0,high = nums.size()-1,mid = 0;
			while(low<=high){
				mid = partation(nums,low,high);
				if(mid == k-1){
					return nums[mid];
				}else if(mid<k-1){
					low = mid + 1;
				}else{
					high = mid -1;
				}
			}
			return -1;
		}
		int partation(vector<int>& nums,int low,int high){
			int left = low+1;
			int right = high;
			swap(nums[low],nums[(low+high)/2]);
			int bound = nums[low];
			while(left<=right){
				while(left<high && nums[left] >= bound) left++;
				while(nums[right]<bound) right--;
				if(left<right){
					swap(nums[left++],nums[right--]);
				}else{
					break;
				}
			}
			swap(nums[low],nums[right]);
			return right;
		}
};

刷题215. Kth Largest Element in an Array

标签:元素   swap   art   line   int   小根堆   cout   auto   iss   

原文地址:https://www.cnblogs.com/siweihz/p/12287120.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!