top K问题的解法

以求最大的第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  有换行的. 不愿搞图片了...
 }