最小的k个数

题目大致为：
输入n个整数，找出其中最小的k个数。
思路：
使用类似二叉查找树的形式实现，控制好树中的结点个数为k。
Java代码

package com.sun.jojo.datastructure;import java.util.Iterator;import java.util.TreeSet;/** * Author sunjiamin * Date 2017/8/15 16:10 * Describe : */public class MinK {    public static void main(String args[]) {        // 测试的例子        int array[] = { 4, 5, 1, 6, 2, 7, 3, 8 };        final int k = 4;        TreeSet<Integer> set = getLeastNumbers(array, k);        // 输出        Iterator<Integer> it = set.iterator();        System.out.println("最小的" + k + "个数为：");        while (it.hasNext()) {            System.out.print(it.next() + "、");        }    }    /**     * TreeSet 特性 如果元素具备自然顺序 的特性，那么就按照元素自然顺序的特性进行排序存储。     *  存放进的数据会自动排好序     * @param array     * @param k     * @return     */    public static TreeSet<Integer> getLeastNumbers(int array[], int k) {        TreeSet<Integer> set = new TreeSet<Integer>();        // 判断k和array的合法性        if (array == null || k <= 0) {            return null;        }        for (int i = 0; i < array.length; i++) {            if (set.size() < k) {// 如果TreeSet中的元素小于K个，则直接插入                set.add(array[i]);            } else {// TreeSet中的元素大于K个                if (set.last() > array[i]) {// 最大的元素大于array[i]                    set.pollLast();// 移除                    set.add(array[i]);// 加入新的                }            }        }        return set;    }}

结果：

最小的4个数为：1、2、3、4、

这里稍微看看TreeSet的add内部实现：

public V put(K key, V value) {        Entry<K,V> t = root;        if (t == null) {            compare(key, key); // type (and possibly null) check            root = new Entry<>(key, value, null);            size = 1;            modCount++;            return null;        }        int cmp;        Entry<K,V> parent;        // split comparator and comparable paths        Comparator<? super K> cpr = comparator;        if (cpr != null) {            do {                parent = t;                cmp = cpr.compare(key, t.key);                if (cmp < 0)                    t = t.left;                else if (cmp > 0)                    t = t.right;                else                    return t.setValue(value);            } while (t != null);        }        else {            if (key == null)                throw new NullPointerException();            @SuppressWarnings("unchecked")                Comparable<? super K> k = (Comparable<? super K>) key;            do {                parent = t;                cmp = k.compareTo(t.key);                if (cmp < 0)                    t = t.left;                else if (cmp > 0)                    t = t.right;                else                    return t.setValue(value);            } while (t != null);        }        Entry<K,V> e = new Entry<>(key, value, parent);        if (cmp < 0)            parent.left = e;        else            parent.right = e;        fixAfterInsertion(e);        size++;        modCount++;        return null;    }

调用了cpr.compare(key, t.key);进行排序