描述
有一个整数数组,请你根据快速排序的思路,找出数组中第 k 大的数。
给定一个整数数组 a ,同时给定它的大小n和要找的 k ,请返回第 k 大的数(包括重复的元素,不用去重),保证答案存在。
要求:时间复杂度 O(nlogn)O(nlogn),空间复杂度 O(1)O(1)
数据范围:0\le n \le 10000≤n≤1000, 1 \le K \le n1≤K≤n,数组中每个元素满足 0 \le val \le 100000000≤val≤10000000
示例1
输入:
[1,3,5,2,2],5,3
复制返回值:
2
复制
示例2
输入:
[10,10,9,9,8,7,5,6,4,3,4,2],12,3
复制返回值:
9
复制说明:
去重后的第3大是8,但本题要求包含重复的元素,不用去重,所以输出9
import java.util.*;
public class Solution {
public int findKth(int[] a, int n, int K) {
quickSort(a,0,n-1,n-K);
return a[n-K];
}
public void quickSort(int []a,int left,int right,int K){
if(left>=right)return ;
int cur=partion(a,left,right);
if(cur==K)return ;
if(cur>K) quickSort(a,left,cur-1,K);
else quickSort(a,cur+1,right,K);
}
public int partion(int[]a,int left,int right){
int start=left;
int temp=a[start];
if(left>=right)return left;
while(left<right){
while (left<right&&temp<=a[right])--right;//非必要时使用左自增
while (right>left&&temp>=a[left])++left;
//交换
if(left<right){
int t=a[right];
a[right]=a[left];
a[left]=t;
}
}
a[start]=a[left];
a[left]=temp;
return left;
}
}