0
点赞
收藏
分享

微信扫一扫

#yyds干货盘点# 面试必刷TOP101:寻找第K大

1.简述:

描述

有一个整数数组,请你根据快速排序的思路,找出数组中第 k 大的数。

给定一个整数数组 a ,同时给定它的大小n和要找的 k ,请返回第 k 大的数(包括重复的元素,不用去重),保证答案存在。

要求:时间复杂度 ,空间复杂度 

数据范围:, ,数组中每个元素满足 

示例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

2.代码实现:

import java.util.*;

public class Solution {
public int findKth(int[] a, int n, int K) {
return quickSort(a, 0, a.length - 1, K);
}

private int quickSort(int[] arr, int left, int right, int k){
int p = partition(arr, left, right);
// 改进后,很特殊的是,p是全局下标,只要p对上topK坐标就可以返回
if (p == arr.length - k) {
return arr[p];
}else if (p < arr.length - k) {
// 如果基准在左边,这在右边找
return quickSort(arr, p + 1, right,k);
}else {
return quickSort(arr, left, p - 1,k);
}
}

private int partition(int[] arr, int left, int right) {
// 可优化成随机,或中位数
int key = arr[left];
while (left < right) {
while (left < right && arr[right] >= key) right--;
arr[left] = arr[right];
while (left < right && arr[left] <= key) left++;
arr[right] = arr[left];
}
arr[left] = key;
return left;
}
}

举报

相关推荐

0 条评论