首页 > 学院 > 开发设计 > 正文

LeetCode 215. Kth Largest Element in an Array

2019-11-06 07:02:08
字体:
来源:转载
供稿:网友

Find the kth largest element in an unsorted array. Note that it is the kth largest element in the sorted order, not the kth distinct element.

For example,Given [3,2,1,5,6,4] and k = 2, return 5.

Note: 

You may assume k is always valid, 1 ≤ k ≤ array's length.

如题所示,求未排序的数组中最大的元素。

第一个想法是使用快排,快排的思想是“挖坑”和“分治”,主要有以下步骤:

1.选定一个位置(如初始位置),将该位置的数“挖”出来,保存到临时变量temp中,这样就留下了第一个坑。

2.从后往前找,看是否有小于temp的数,如果有就将其放入之前的坑中,此时留下了第二个坑。

3.再从前往后找,看是否有大于temp的数,如果有就将其放入第二个坑中。

接下来重复步骤2和步骤3,直到左右两边的下标相遇。

4.再将temp放入下标中。

这样就得到了以初始的数为中心,左边比它小,右边比它大的序列。

5.最后,只需要对L~i-1和i+1~R两部分递归就行。

由于我们是寻找第K大的数,因此只需要判断这个数是属于L~i-1还是i+1~L,因此只需要对一半的数字进行排序,这样就能大大地简化步骤。

代码实现如下:

void quick_sort(vector<int> &nums,int l,int r,int k){	int i=l,j=r;	if(i<j){		int temp = nums[l];		while(i < j){			while(i < j && temp <= nums[j])				j--;	    	if(i<j){	    		nums[i] = nums[j];	    		i++;	    	}	    	while(i<j &&temp >= nums[i])	    		i++;	    	if(i<j){	    		nums[j] = nums[i];	    		j--;	    	}	    }	    nums[i] = temp;	    if(i == nums.size()-k)	    	return ;	    else if(i<nums.size()-k)	    	quick_sort(nums,i+1,r,k);	    else	    	quick_sort(nums,l,i-1,k);	  }}class Solution {public:    int findKthLargest(vector<int>& nums, int k) {        quick_sort(nums,0,nums.size()-1,k);    	return nums[nums.size()-k];    }};


发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表