题目链接:
https://leetcode.cn/problems/kth-largest-element-in-an-array/description/?envType=study-plan-v2&envId=top-100-liked
参考的K神题解:
https://leetcode.cn/u/jyd/
题目
给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。
请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
你必须设计并实现时间复杂度为 O(n) 的算法解决此问题。
难度
中等,虽然标的是中等难度,但是我写了好久,555~
思路:
下面的思路是K神的:
作者:Krahets
链接:https://leetcode.cn/problems/kth-largest-element-in-an-array/solutions/2361969/215-shu-zu-zhong-de-di-k-ge-zui-da-yuan-d786p/
使用递归法,递归快排实现。
快速排序的核心包括“哨兵划分” 和 “递归” 。
哨兵划分: 以数组某个元素(一般选取首元素)为基准数,将所有小于基准数的元素移动至其左边,大于基准数的元素移动至其右边。
递归: 对 左子数组 和 右子数组 递归执行 哨兵划分,直至子数组长度为 1 时终止递归,即可完成对整个数组的排序。
图解:
同样来自K神,我只是搬运一下方便自己回顾整理。
代码:
class Solution {
public int findKthLargest(int[] nums, int k) {
LinkedList list=new LinkedList<>();
for(int i=0;i<nums.length;i++){
list.add(nums[i]);
}
return quickSelect(list,k);
}
public int quickSelect(LinkedList<Integer> nums,int k){
LinkedList<Integer> big=new LinkedList<>();
LinkedList<Integer> equal=new LinkedList<>();
LinkedList<Integer> small=new LinkedList<>();
int temp=nums.get(0);
for(Integer i:nums){
if(i==temp) equal.add(i);
else if(i<temp) small.add(i);
else big.add(i);
}
if(k<=big.size()) return quickSelect(big,k);
else if(k>(big.size()+equal.size())){
return quickSelect(small,k-(big.size()+equal.size()));
}else return temp;
}
}