给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。
请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。输入: [3,2,1,5,6,4] 和 k = 2 输出: 5
代码3 —— 使用快排中的分治思想
重点在于 partition 函数,通过 partition 操作来定位数组中索引为 k - 1 的元素。
在主函数中,需要不断调整分治的位置,直到 position = k - 1 。
- while(true)就是无限循环语句。所以代码中就会一直循环,直到在第一个if语句中return。
- quicksort函数:把以pivot=num[low]为界,分割成前大后小的两部分,最后return pivot的位置(low=high)
- getkth函数:看quicksort返回的pivot的位置是不是第k个,k-1,否则变换low和high
class Solution {
public int findKthLargest(int[] nums, int k) {
return getkth(nums,k);
}
int quicksort(int nums[],int low,int high){
int pivot=nums[low];//把nums[low]暂存起来
while(low<high){
while(low<high&&nums[high]<pivot){
high--;//从后边开始找(因为暂存的是前边的数)满足比pivot小的,直接跳过
}
nums[low]=nums[high];//不满足 交换
while(low<high&&nums[low]>=pivot){//记得等于
low++;
}
nums[high]=nums[low];
}
nums[low]=pivot;//high和low是同一个节点了,即中点,把pivot放进去
return low;//返回中间位置
}
int getkth(int nums[],int k){
int low=0;
int high=nums.length-1;
// 不断调整分治的位置,直到 position = k - 1
while(true){//为了结束递归and避免递归前边的初始化
int index =quicksort(nums,low,high);//需要改变边界值
if(index==k-1){
return nums[index];
}
else if(index<k-1){
low=index+1;
}
else{
high=index-1;
}
}
}
}