一、题目描述
二、整体思路
(1)快排
对于SELECT K问题,可以通过三路快排解决,快排可以把一个元素放至按升序排序的数组正确的位置,左边为小于该元素的元素集合,右边为大于该元素的元素集合。
三路快排把数组分成三部分:
[l,l2-1]:<nums[l]的元素区间;[l2,r2-1]:=nums[l]的元素区间;[r2,r]:>nums[l]的元素区间;
当k位于[l2,r2-1]区间时,说明数组中[l2,r2-1]区间的元素都放置到正确的位置,k在其中说明k所指的元素已经排序至正确位置,可以返回。
当k位于[l,l2-1]区间时,说明还需要对[l,l2-1]进行快速排序,同理当k位于[r2,r]区间时,说明要对[r2,r]进行快速排序。
(2)小根堆
要选择第K个最大元素,则可以把该数组[0,k-1]的部分转化为小根堆,遍历[k,nums.length-1]的部分,若遍历到的元素大于堆顶元素,则可以交换此两元素,然后对新的堆顶元素进行siftDown下沉操作,重复上述步骤,最终可以得到前K个最大元素组成的小根堆,堆顶即为第K大的元素。
(3)原地堆排序、大根堆
对整个数组进行堆排序形成大根堆。把堆顶元素与数组末尾元素进行交换,这样可以得到第一大的元素。把新的堆顶元素在[0,nums.length-1)处进行siftdown()下沉操作,此时的堆顶元素即为数组第二大的元素,那么又可以与数组的[nums.length-2]元素进行交换得到新的堆。重复上述步骤直到得到数组第K大的元素。
三、代码
//快排
class Solution {
public int findKthLargest(int[] nums, int k) {
Random random=new Random();
quicksort(nums,0,nums.length-1,nums.length-k,random);
return nums[nums.length-k];
}
public void quicksort(int[] nums,int l,int r,int k2,Random random){
if(l>=r) return;
//把nums[l]与数组中随机位置的元素进行交换
//防止快排退化导致时间复杂度增加
int p=random.nextInt(r-l+1)+l;
int temp3=nums[l];
nums[l]=nums[p];
nums[p]=temp3;
int i=l+1,l2=l,r2=r+1;
while(i<r2){
//[l,l2]:<nums[l]元素区间;[l2+1,i-1]:=nums[l]元素区间;[r2,r]:>nums[l]//元素区间
if(nums[i]<nums[l]){
int temp=nums[++l2];
nums[l2]=nums[i];
nums[i]=temp;
i++;
}else if(nums[i]>nums[l]){
int temp4=nums[--r2];
nums[r2]=nums[i];
nums[i]=temp4;
}else{
i++;
}
}
//最后把nums[l]与nums[l2]交换
int temp2=nums[l];
nums[l]=nums[l2];
nums[l2]=temp2;
//区间发生变化。[l,l2-1]:<;[l2,r2-1]:=;[r2,r]:>
if(k2>=l2 && k2<=r2-1) return;
else if(r2<=k2) quicksort(nums,r2,r,k2,random);
else quicksort(nums,l,l2-1,k2,random);
}
}
//小根堆
class Solution {
public int findKthLargest(int[] nums, int k) {
//把数组[0,k-1]的部分化为小根堆
for(int i=(k-2)/2;i>=0;i--){
siftDown(nums,i,k);
}
//遍历数组剩下部分,同时把小根堆堆顶替换为更大的数组元素并对新堆顶执行下沉操作
for(int i=k;i<nums.length;i++){
if(nums[i]>nums[0]){
int temp=nums[i];
nums[i]=nums[0];
nums[0]=temp;
siftDown(nums,0,k);
}
}
return nums[0];//此时小根堆代表前K大的元素,堆顶表示第K大的元素
}
public void siftDown(int[] arr,int i,int n){
while((i*2+1)<n){
int j=i*2+1;
if(j+1<n && arr[j+1]<arr[j]) j++;
if(arr[i]>arr[j]){
int temp2=arr[i];
arr[i]=arr[j];
arr[j]=temp2;
i=j;
}else break;
}
}
}
//堆排序,大根堆
class Solution {
public int findKthLargest(int[] nums, int k) {\
//把整个数组化为大根堆
for(int i=(nums.length-2)/2;i>=0;i--){
siftDown(nums,i,nums.length);
}
int ret=0;
//从数组末尾由后到前遍历,交换堆顶元素,把元素从最大元素到最小元素的顺序排序
for(int i=nums.length-1,k2=0;i>=0;i--){
swap(nums,0,i);
siftDown(nums,0,i);
k2++;
if(k2==k) return nums[i];
}
return ret;
}
public void swap(int[] nums,int i,int j){
int temp=nums[i];
nums[i]=nums[j];
nums[j]=temp;
}
public void siftDown(int[] nums,int i,int n){
while((i*2+1)<n){//存在左子结点时
int j=i*2+1;
if(j+1<n && nums[j+1]>nums[j]) j++;//存在右子节点且右子节点比左子节点更大时
if(nums[i]<nums[j]){//较小的儿子节点比当前节点大时,要下沉当前节点
swap(nums,i,j);
i=j;
}else break;
}
}
}