面试题：求第K大元素（topK）【加强版】

在原来基础上增长了算法E。

1、引言

​ 这就是相似求Top（K）问题，什么意思呢？怎么在无序数组中找到第几（K）大元素？咱们这里不考虑海量数据，能装入内存。

2、普通算法

算法A：

将数组中的元素升序排序，找到数组下标k-1的元素便可。这是你们最容易想到的方法，若是使用简单排序算法，时间复杂度为O(n^2)。

算法B:

1. 第一步：初始化长度为K的一个数组，先读入K个元素，将元素降序排序（升序也能够），这时候第K大元素就在最后一个。
2. 第二步：读入下一个元素就与已排序的第K大元素比较，若是大于，则将当前的第K大元素删掉，并将此元素放到前K-1个正确的位置上（这里就是简单的插入排序了。不了解插入排序的朋友能够看这里图解选择排序与插入排序）。
3. 时间复杂度：第一步采用普通排序算法时间复杂度是O(k^2)；第二步：(N-k)*(k-1) = Nk-k^2+k。因此算法B的时间复杂度为O(NK)。当k=N/2（向下取整）时，时间复杂度仍是O(n^2)。

其实求第K大问题，也能够求反，即求第N-k+1小问题。这是等价的。因此当K=N/2时，是最难的地方，但也颇有趣，这时候的K对应的值就是中位数。

3、较好算法

算法C：

1. 算法思想：将数据读入一个数组，对数组进行buildHeap(咱们这里构建大顶堆)，以后对堆进行K次deleteMax操做，第K次的结果就是咱们须要的值。（由于是大顶堆，因此数据从大到小排了序，堆排序之后会详细说）。

2. 如今咱们来解决上节遗留的问题，为何buildHeap是线性的？不熟悉堆的能够看一下 图解优先队列（堆）。咱们先来看看代码实现。

   public PriorityQueue(T[] items) {
          //当前堆中的元素个数
          currentSize = items.length;
          //可自行实现声明
          array = (T[]) new Comparable[currentSize +1];
          int i = 1;
          for (T item : items){
              array[i++] = item;
          }
          buildHeap();
      }
   
   private void buildHeap() {
          for (int i = currentSize / 2; i > 0; i--){
              //堆的下滤方法，可参考上面的连接
              percolateDown(i);
          }
      }
   复制代码

图中初始化的是一颗无序树，通过7次percolateDown后，获得一个大顶堆。从图中能够看到，共有9条虚线，每一条对应于2次比较，总共18次比较。为了肯定buildHeap的时间界，咱们须要统计虚线的条数，这能够经过计算堆中全部节点的高度和获得，它是虚线的最大条数。该和是O(N)。

定理：包含2^h+1-1个节点、高为h的理想二叉树（满二叉树）的节点的高度的和是2^h+1-1-（h+1）。

1. 什么叫满二叉树？满二叉树是彻底填满的二叉树，最后一层都是填满的，如图中所示。彻底二叉树，是除最后一层之外都是填满的，最后一层外也必须从左到右依次填入，就是上一篇中说的堆的结构。满二叉树必定是彻底二叉树，彻底二叉树不必定是满二叉树。

2. 证实定理：

   容易看出，满二叉树中，高度为h上，有1个节点；高度h-1上2个节点，高度h-2上有2^2个节点及通常在高度h-i上的2ⁱ个节点组成。
   

方程两边乘以2获得：

两式相减获得：

因此定理得证。由于堆由彻底二叉树构成，因此堆的节点数在2^h和2^h+1之间，因此意味着这个和是O(N)。因此buildHeap是线性的。因此算法C的时间复杂度是：初始化数组为O(N)，buildHeap为O(N)，K次deleeMax须要O(klogN)，因此总的时间复杂度是：O(N+N+klogN)=O(N+klogN)，若是K为N/2时，运行时间是O(NlogN)。

算法D：

1. 算法思想：咱们采用算法B的思想，只是咱们这里构建一个节点数为K的小顶堆，只要下一个数比根节点大，就删除根节点，并将这个数进行下滤操做。因此算法最终的第K大数就是根节点的数。
2. 时间复杂度：对K个数进行buildHeap是O(k)，最坏状况下假设剩下的N-k个数都要进入堆中进行下滤操做，总的须要O(k+(N-k)logk)。若是K为N/2，则须要O(NlogN)。

算法E：快速选择

参考快速排序及其优化，使用快速排序的思想，进行一点点改动。

• 算法思想：

1. 若是S中元素个数是1，那么k=1并将S中的元素返回。若是正在使用小数组的截止方法且|S|<=CUTOFF，则将S排序并返回第K个最大元素
2. 选取一个S中的元素v，称之为枢纽元（pivot）；
3. 将S-{v}（S中除了枢纽元中的其他元素）划分为两个不相交的集合S₁和S₂，S₁集合中的全部元素小于等于枢纽元v，S₂中的全部元素大于等于枢纽元；
4. 若是k<=|S₁|，那么第k个最大元必然在S₁中。这时，咱们返回quickSelect(S₁，k)。若是k=1+|S₁|，那么枢纽元就是第k个最大元，咱们直接返回枢纽元。不然，这第k个最大元必定在S₂中，它就是S₂中的第（k-|S₁|-1）个最大元。咱们进行一次递归调用并返回quickSelect(S₂，k-|S₁|-1)。

• java实现：

public class QuickSelect {
    /**
     * 截止范围
     */
    private static final int CUTOFF = 5;

    public static void main(String[] args) {
        Integer[] a = {8, 1, 4, 9, 6, 3, 5, 2, 7, 0, 12, 11, 15, 14, 13, 20, 18, 19, 17, 16};
        int k = 5;
        quickSelect(a, a.length - k + 1);
        System.out.println("第" + k + "大元素是：" + a[a.length - k]);
    }

    public static <T extends Comparable<? super T>> void quickSelect(T[] a, int k) {
        quickSelect(a, 0, a.length - 1, k);
    }

    private static <T extends Comparable<? super T>> void quickSelect(T[] a, int left, int right, int k) {
        if (left + CUTOFF <= right) {
            //三数中值分割法获取枢纽元
            T pivot = median3(a, left, right);

            //开始分割序列
            int i = left, j = right - 1;
            for (; ; ) {
                while (a[++i].compareTo(pivot) < 0) {
                }
                while (a[--j].compareTo(pivot) > 0) {
                }
                if (i < j) {
                    swapReferences(a, i, j);
                } else {
                    break;
                }
            }
            //将枢纽元与位置i的元素交换位置
            swapReferences(a, i, right - 1);

            if (k <= i) {
                quickSelect(a, left, i - 1, k);
            } else if (k > i + 1) {
                quickSelect(a, i + 1, right, k);
            }
        } else {
            insertionSort(a, left, right);
        }
    }

    private static <T extends Comparable<? super T>> T median3(T[] a, int left, int right) {
        int center = (left + right) / 2;
        if (a[center].compareTo(a[left]) < 0) {

            swapReferences(a, left, center);
        }
        if (a[right].compareTo(a[left]) < 0) {
            swapReferences(a, left, right);
        }
        if (a[right].compareTo(a[center]) < 0) {
            swapReferences(a, center, right);
        }
        // 将枢纽元放置到right-1位置
        swapReferences(a, center, right - 1);
        return a[right - 1];
    }

    public static <T> void swapReferences(T[] a, int index1, int index2) {
        T tmp = a[index1];
        a[index1] = a[index2];
        a[index2] = tmp;
    }

    private static <T extends Comparable<? super T>> void insertionSort(T[] a, int left, int right) {
        for (int p = left + 1; p <= right; p++) {
            T tmp = a[p];
            int j;

            for (j = p; j > left && tmp.compareTo(a[j - 1]) < 0; j--) {
                a[j] = a[j - 1];
            }

            a[j] = tmp;
        }
    }
}

//输出结果
//第5大元素是：16
复制代码

由于我这里是升序排序，因此求第K大，与求第N-k+1是同样的。

• 最坏时间复杂度：与快速排序同样，当一个子序列为空时，最坏为O(N²)。
• 平均时间复杂度：能够看出，快速选择每次只用递归一个子序列。平均时间复杂度为O(N)。

4、总结

本篇详述了 求top(K)问题的几种解法，前两种十分平凡普通，后两种比较优一点，暂时给出求解中位数须要O(NlogN)时间。后面介绍使用快速选择方法，每次只用递归一个子序列，能够达到平均O(N)时间复杂度。