Java集合之TreeSet

阅读目录

• 一、概述
• 二、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个元素，那么平均需要进行log₂n次比较，才能找到新元素的正确位置。