Java TreeSet 树集合

---

TreeSet

• TreeSet 类继承了 Set 接口和 SortedSet 接口，元素不能重复；
• 由于 SortedSet 接口能对元素升序排序（也叫自然排序），TreeSet 类也会对实现了 Comparable 接口 的类的对象自动排序，或者根据创建 TreeSet 时提供的 Comparator 进行排序；
• 底层依赖于 TreeMap，元素没有索引；
• 通过红黑树实现，不允许放入null值；
• 存储大量的需要进行快速检索的排序信息时，TreeSet 是一个很好的选择；
• java.util.TreeSet，详见：Class TreeSet；

---

1. 构造方法

构造方法	说明
TreeSet()	构造一个空的树集合
TreeSet(Collection c)	构造一个包含了集合 c 的树集合
TreeSet(Comparator< E> comparator)	构造一个空的树集合，按照由 comp 指定的比较方法进行排序
TreeSet(SortedSet ss)	构造一个包含了已排序集合 ss 的元素的树集合

---

2. 提供的方法

类型	方法	说明
E	first()	返回集合中的最低元素，数据类型为 E
E	last()	返回集合中的最高元素，数据类型为 E
E	pollFirst()	返回并删除最低元素，数据类型为 E
E	pollLast()	返回并删除最高元素，数据类型为 E
SortedSet<>	subSet(E fromElement, E toElement)	返回一个新集合，包括 fromElement，不包括 toElement（即 [ ) ）
SortedSet<>	tailSet(E fromElement)	返回一个新集合，包括 fromElement
SortedSet<>	headSet(E toElement)	返回一个新集合，不包括 toElement

---

3. 类的使用

a. 自然排序

import java.util.Iterator;
import java.util.SortedSet;
import java.util.TreeSet;

public class test {
    public static void main(String[] args) {
        TreeSet ts = new TreeSet();
        ts.add(3);
        ts.add(4);
        ts.add(2);
        ts.add(1);
        System.out.println(ts);
        System.out.println(ts.size());
        Iterator it = ts.iterator();
        while (it.hasNext()) {
            System.out.print(it.next() + " ");//1 2 3 4
        }
        System.out.println();
        System.out.print(ts.first() + " ");//1
        System.out.print(ts.last() + " ");//4
        System.out.print(ts.pollFirst() + " ");//1
        System.out.print(ts.pollLast() + " ");//4
        System.out.println();
        SortedSet ss = ts.subSet(2, 3);
        SortedSet ss2 = ts.headSet(3);
        SortedSet ss3 = ts.tailSet(4);
        it = ss.iterator();
        while (it.hasNext()) {
            System.out.print(it.next() + " ");//2
        }
    }
}
/*
输出
[1, 2, 3, 4]
4
1 2 3 4 
1 4 1 4 
2 
 */

b. 比较器排序

import java.util.Comparator;
import java.util.TreeSet;

public class test {
    public static void main(String[] args) {
        //有参构造,传入比较器,使用比较器对元素进行排序
        TreeSet<Integer> set = new TreeSet<Integer>(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                //元素前 - 元素后 : 升序
                //元素后 - 元素前 : 降序
                return o2 - o1;
            }
        });
        set.add(20);
        set.add(18);
        set.add(23);
        set.add(22);
        set.add(17);
        set.add(24);
        set.add(19);
        System.out.println(set);

    }
}
/*
输出
[24, 23, 22, 20, 19, 18, 17]
 */