张小飞的Java之路——第三十三章——Set——TreeSet

写在前面：

视频是什么东西，有看文档精彩吗？

视频是什么东西，有看文档速度快吗？

视频是什么东西，有看文档效率高吗？

诸小亮：下面我们介绍 Set 的另一个实现类——TreeSet

张小飞：这个跟 HasSet 有什么区别？

诸小亮：最大的区别是——存储数据的方式不同

特点

诸小亮：我们先看看 TreeSet 独有的一些特点

去重且排序

诸小亮：使用 TreeSet 不仅会自动去重，还会对集合中的元素自动排序

张小飞：排序？

诸小亮：是的，来看下面的代码

public static void main(String[] args) throws Exception {
    // TreeSet 可以去重和排序
    Set set = new TreeSet();
    set.add(3);
    set.add(1);
    set.add(2);
    set.add(1);
    System.out.println(set);
}

结果：，

张小飞：它排序的原理是什么呢？

诸小亮：它默认根据元素的自然顺序排序

张小飞：什么是自然顺序？

诸小亮：自然顺序，就是 ASCII 码表中的顺序，根据它们在 ASCII 码表中的值进行排序

张小飞：明白了

元素必须实现 Comparable 接口

诸小亮：另外，TreeSet 存储的元素必须实现 Comparable 接口

张小飞：必须吗？

诸小亮：是的，下面代码 Hero 没有就实现 Comparable

public static void main(String[] args) throws Exception {
    Set set = new TreeSet();
    set.add(new Hero("妲己"));
    set.add(new Hero("西施"));
    set.add(new Hero("嫦娥"));
    set.add(new Hero("妲己"));
    System.out.println(set);
}

但是其结果：

张小飞：明白了，不过这个 Comparable 接口是干嘛的？

Comparable

诸小亮：Comparable——是一个排序接口，该接口中只有一个 compareTo 方法，用来比较对象大小

诸小亮：如果一个类实现了Comparable接口，就意味着该类的对象支持排序

张小飞：TreeSet就是根据这个特性，才能对元素进行排序的吗？

诸小亮：没错，我们修改 Hero 实现 Comparable，复写 compareTo 方法

@Override
public int compareTo(Object obj) {
    if(obj instanceof Hero){
        //String 已经有 compareTo 方法，可以直接使用
        return this.getName().compareTo(((Hero)obj).getName());
    }
    throw new IllegalArgumentException("obj不是Hero对象");
}

测试代码：

public static void main(String[] args) throws Exception {
    Hero xishi = new Hero("西施");
    Hero change = new Hero("嫦娥");
    System.out.println(xishi.compareTo(change));
}

结果：

张小飞：这个 11929 是？？？

诸小亮：compareTo 返回值是 int，可以分 3 类：

• 负数：表示当前对象 < 指定对象，xishi < change
• 正数：表示当前对象 > 指定对象，xishi > change
• 0：表示两个对象相等

张小飞：原来如此

二叉树

张小飞：您刚才说 TreeSet 的跟 HashSet 的底层结构不同

诸小亮：是的，TreeSet对的底层是二叉树

张小飞：二叉树？

诸小亮，二叉树——也是一种数据结构，如下图：

数据在内存中的存储方式是树形结果，比如：

张小飞：能具体解释一下存储数据的过程吗？

诸小亮：当然可以，其大致的存储过程是

• 第一次存的 5 是顶节点
• 第二次存 8，放到 5 的右侧
• 第三次存 3， 放到 5 的左侧
• 第三次存 6， 放到 5 的左侧，又比 8 小，放到 8 的右侧
• 。。。。。

TreeSet 就是这样排列元素的顺序

红黑树

张小飞：按照上面的存储过程，如果是这样添加数据，那么是不是就成链表了

Set set = new TreeSet();
set.add(1); set.add(2); set.add(3); 

按照上面的方式，最终的结构是：

诸小亮：这个问题非常棒，其实 TreeSet 的底层结构是——红黑树

• 红黑树——是一种自平衡二叉搜索树

张小飞：自平衡是什么意思？

诸小亮：就是可以通过自旋转，保证平衡，看下图

张小飞：原来如此，您这是一个网站？

诸小亮：是的，地址是：https://www.cs.usfca.edu/~galles/visualization/RedBlack.html

张小飞：好的，我也来试试

保证元素唯一

张小飞：TreeSet 能够去重，是不是就是因为实现了 compareTo 方法？

诸小亮：是的，TreeSet 添加元素时，会调用已有元素的 compareTo 方法判断添加的元素是否已经存在，但是也可以不实现 Comparable 接口

张小飞：嗯？不是必须实现吗？刚才没有实现都报错了

诸小亮：如果没有实现 Comparable 接口，可以自定义比较器

张小飞：这是什么意思？

自定义比较器

诸小亮：工作中有这种情况，需要排序，但是实体类中没有排序功能，而且也不能修改实体类的代码

• 注意：工作中不是自己的代码不能随意修改，否则可能产生意想不到的问题

张小飞：这时候，就用你说的——自定义比较器 ？

诸小亮：是的，自定义比较器——让集合对象具备排序功能

张小飞：具体怎么使用呢？

诸小亮：TreeSet 有一个构造函数支持传递一个 Comparator 类型的对象

张小飞：Comparator 是什么？？？

诸小亮：Comparator——比较器接口，它有一个 compare 方法，支持比较两个对象的大小

下面是一个示例，你来看看吧

//注意：Hero 没有实现 Comparable 接口
class Hero{
    String name;

    public Hero(String name) {
        this.name = name;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public boolean equals(Object obj){
        if(obj instanceof Hero){
            return this.getName().equals(((Hero)obj).getName());
        }
        return false;
    }

    @Override
    public int hashCode() {
        return name.hashCode();
    }

    @Override
    public String toString() {
        return "Hero{" +
                "name='" + name + '\'' +
                '}';
    }

}

public class StringDemo {
    public static void main(String[] args) throws Exception {
        //1. 使用匿名内部类创建一个比较器对象
        Comparator comparator = new Comparator() {
            @Override
            public int compare(Object o1, Object o2) {
                Hero h1 = (Hero) o1;
                Hero h2 = (Hero) o2;
                return h1.getName().compareTo(h2.getName());
            }
        };
        //2. 创建对象时，把比较器对象传进去
        Set set = new TreeSet(comparator);
        //3. 这时Hero没有比较功能也不影响排序
        set.add(new Hero("daji"));
        set.add(new Hero("xishi"));
        set.add(new Hero("change"));
        System.out.println(set);
    }
}

结果：

张小飞：原来是这样，现在是升序，如果想要倒序呢？

诸小亮：这简单，修改 compare 方法就行了，比如：

结果：

6. 轻微总结

诸小亮：来，我们总结一下，集合框架的最顶层接口时 Collection，其下有两个子接口：List、Set
List 有序集合，主要有 3 个具体实现类：

• ArrayList：底层可变数组，自动扩容(原来的1.5倍)，允许存重复元素，多线程下不安全
  • 插入删除指定元素速度慢，根据下标获取元素快
• LinkedList：底层链表，允许存重复元素，多线程下不安全
  • 插入删除指定元素速度快，根据下标获取元素慢
• Vector：底层可变数组，都是同步方法，ArrayList 是它的替代品

Set 无序集合，主要有 2 个具体实现类：

• HashSet：自动去重，底层哈希表
• TreeSet：自动去重，自动排序，底层红黑树结构
  • 需要元素自身有比较功能，默认根据自然顺序排序
  • 如果元素没有比较功能，需要集合有排序功能（自定义比较器）