寻找第K大
有一个整数数组,请你根据快速排序的思路,找出数组中第K大的数。 给定一个整数数组a,同时给定它的大小n和要找的K(K在1到n之间),请返回第K大的数。
测试样例:
[1,3,5,2,2],5,3
返回:2
分析:
根据快速排序的思路:
快排参见:https://blog.youkuaiyun.com/qq_43109561/article/details/90235946
按如下步骤在n个元素里面第K大的元素:
第一步:进行一次快排(将大的元素放在前半段,小的元素放在后半段),得到的中轴p
第二步:判断 p - low + 1 == k ,如果成立,直接返回a[p],(前半段有p - low(等于k-1)个大于a[p]元素,则a[p]为第K大的元素)
第三步:如果 p - low + 1 > k,则第k大的元 素在前半段,更新high = p - 1,执行第一步
第四步:如果 p - low + 1 < k,则第k大的元素在后半段,更新low = p + 1, 且 k = k - (p - low + 1),执行第一步常规快排得到整体有序的数组复杂度是O(nlgn),此方法每次可以去掉“一半”的元素,复杂度为O(n)。
代码:
public class Finder {
public int findKth(int[] a, int n, int K) {
return quickSort