以求最大的第K个数为例
共提供3种实现分别为
1.排序 时间复杂度 O(nlgn)
2.利用最小堆(把最小的都过滤了, 大的都沉底了。) 时间复杂度 O(n * lgk) 可利用java中的优先队列. 底层是二叉堆.
3.快速排序中的划分 时间复杂度 O(n)
public static int KthLargest(int[] a, int k) {
Arrays.sort(a);
return a[a.length - k];
}
public static int KthLargestII(int[] a, int k) {
PriorityQueue<Integer> pq = new PriorityQueue<>();
for(int i = 0; i < a.length; i++) {
pq.offer(a[i]);
if(pq.size() > k) {
pq.poll();
}
}
return pq.peek();
}
public static void swap(int[] a, int i, int j) {
int tmp = a[i];
a[i] = a[j];
a[j] = tmp;
}
public static int KthLargestIII(int[] a, int k) {
k = a.length - k;
int lo = 0;
int hi = a.length - 1;
while (lo < hi) {
int j = partition(a, lo, hi);
if (j < k) {
lo = j + 1;
} else if (j > k) {
hi = j - 1;
} else {
break;
}
}
return a[k];
}
public static int partition(int[] a, int lo, int hi) {
int i = lo;
int j = hi + 1;
int v = a[lo];
while (true) {
while (a[++i] < v) {
if (i == hi) {
break;
}
}
while (a[--j] > v) {
if (j == lo) {
break;
}
}
if (i >= j) break;
swap(a, i, j);
}
swap(a, lo, j);
return j;
}
测试
public static void main(String[] args) {
int[] a = {1,2,3,4,5};
System.out.println(KthLargest(a,1));
System.out.println(KthLargestII(a,1));
System.out.println(KthLargestIII(a,1));
//结果是 5 5 5 有换行的. 不愿搞图片了...
}