NC88 寻找第K大

package NC;

/**
* NC88 寻找第K大
* 有一个整数数组,请你根据快速排序的思路,找出数组中第 k 大的数。
*
* 给定一个整数数组 a ,同时给定它的大小n和要找的 k ,请返回第 k 大的数(包括重复的元素,不用去重),保证答案存在。
* 要求:时间复杂度 O(nlogn),空间复杂度 O(1)
*
* @author Tang
* @date 2021/10/11
*/
public class FindKth {

/**
* 写个快排
* 快排原理—每一组交换为了找到数组最中间的数字
*
* @param a
* @param n
* @param K
* @return
*/
public int findKth(int[] a, int n, int K) {
// write code here
quickSort(a, 0, n-1);
return a[n - K];
}

/**
* 写个快排
* 快排原理: 将数组第一个值当作base,给base找到数组中该有的位置
* 比base小的都放在左边,比base大的都放在右边
* @param a
* @param start
* @param end
*/
private void quickSort(int[] a, int start, int end) {
if(start >= end) {
return;
}
int i = start;
int j = end;
//给base找到合适位置
int base = a[start];
while(i < j) {
while (i < j && base <= a[j]) {
j--;
}

if(i < j && base > a[j]) {
int temp = a[i];
a[i] = a[j];
a[j] = temp;
i++;
}

while (i < j && base >= a[i]) {
i++;
}

while (i < j && base < a[i]) {
int temp = a[i];
a[i] = a[j];
a[j] = temp;
j--;
}
}
a[i] = base;
quickSort(a, start, i-1);
quickSort(a, i+1, end);

}

public static void main(String[] args) {
int[] num = {10,10,9,9,8,7,5,6,4,3,4,2};
int kth = new FindKth().findKth(num, 12, 3);
System.out.println(kth);
}
}
posted @ 2021-10-11 16:44  六小扛把子  阅读(95)  评论(0编辑  收藏  举报