数组中的第K个最大元素

18 篇文章 0 订阅
10 篇文章 0 订阅

题目
在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。

法一:排序后输出。
法二:优先队列(堆)。
法三:利用快速排序的方法。
在这里仅介绍法三,法一法二调用库函数即可。

思路
首先,找第k个最大的元素,即是找第n-k+1个最小的元素。

步骤一
我们利用快排的思想,每一次对一个元素进行归位,判断该元素最终位置p与n-k+1位置关系。
步骤二
1、n-k+1>p:则对p右侧元素进行步骤一操作。
2、n-k+1<p:则对p左侧元素进行步骤一操作。
直到n-k+1 = p停止。

public int findKthLargest(int[] nums, int k) {
		if(nums.length==1) {
			return nums[0];
		}
        helper(nums,0,nums.length-1,nums.length-k);
        return nums[nums.length-k];
    }
	public void helper(int[] nums,int lo,int hi,int k){
		int i = lo;
		int j = hi+1;
		int v = nums[lo];
		while(true) {
			while(i+1<nums.length&&nums[++i]<v) {if(i==hi)break;}
			while(j-1>=0&&nums[--j]>v);
			if(i>=j) {break;}
			int temp = nums[i];
			nums[i] = nums[j];
			nums[j] = temp;
		}
		int temp = nums[j];
		nums[j] = v;
		nums[lo] = temp;
		
		if(j>k) {
			helper(nums,lo,j-1,k);
		}
		else if(j<k) {
			helper(nums,j+1,hi,k);
		}
	}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值