Java集合之TreeSet
一、概述
1、TreeSet(树集)是一个有序集合,可以按照任何顺序将元素插入该集合,当对该集合进行迭代时,各个值将自动以排序后的顺序出现。TreeSet中的元素按照升序排列,缺省是按照自然顺序进行排序,意味着TreeSet中的元素要实现Comparable接口,或者有一个自定义的比较器Comparator。
2、TreeSet底层使用的是TreeMap,TreeMap的底层实现是红黑树(详细见http://www.cnblogs.com/xujian2014/p/4645943.html)。
public TreeSet() { this(new TreeMap<E,Object>()); }
二、TreeSet使用示例
public class Test { public static void main(String[] args) { TreeSet<String> treeSet=new TreeSet<>(); treeSet.add("Bili"); treeSet.add("Amy"); treeSet.add("cDy"); for (String string : treeSet) { System.out.println(string); } }
由于String类实现了Comparable接口,它的compareTo方法是按照字典顺序来对字符串进行排序,所以结果如下:
如果想插入自定义的对象,必须通过实现Comparable接口来定义一个排列的顺序,eg:
class Person implements Comparable<Person> { int age; String name; String address; public Person(int age, String name) { super(); this.age = age; this.name = name; } @Override public int compareTo(Person o) { return this.age-o.age; } } public class TestMain { public static void main(String[] args) { TreeSet<Person> treeSet=new TreeSet<Person>(); treeSet.add(new Person(12, "xujian")); treeSet.add(new Person(6, "luyang")); treeSet.add(new Person(10, "xiewei")); for (Person person : treeSet) { System.out.println(person.name+":"+person.age); } } }
执行结果如下:
当然也可以自定义一个比较器Comparator,实现compare方法。
注意:
1、TreeSet的排列顺序必须是全局顺序,也就是说任何两个元素都是必须可比的,同时只有当他们比较相同时才返回0。
2、如果树集包含了n个元素,那么平均需要进行log2n次比较,才能找到新元素的正确位置。