TreeSet源码解读

TreeSet源码解读

特点: 可以排序

解读

1. 当我们使用无参构造器[new TreeSet();] 创建TreeSet时，仍然是无序的
2. 如果希望添加的元素，按照字符串大小来排序
   2.1.使用TreeSet 提供的一个构造器[ new TreeSet(new Comparator()]，可以传入一个比较器(匿名内部类) 并指定排序规则

看看TreeMap(Comparator<? super K> comparator)的源码

1. 构造器把传入的比较器对象，赋给 TreeSet的底层的 TreeMap的属性this.comparator

    public TreeMap(Comparator<? super K> comparator) {
           this.comparator = comparator;
    }

2. 在 调用 treeSet.add(“tom”), 在底层会执行到 [TreeMap类里的add()方法]

	Comparator<? super K> cpr = comparator; //把之前TreeMap里的比较器赋给cpr
    if (cpr != null) {//cpr 就是我们的匿名内部类(对象)
       do {
           parent = t;
           cmp = cpr.compare(key, t.key); //动态绑定到我们的匿名内部类(对象)compare
           if (cmp < 0)
               t = t.left;
           else if (cmp > 0)
               t = t.right;
           else //如果相等，即返回0,这个Key就没有加入
               return t.setValue(value);
       } while (t != null);
   }

实际应用

    TreeSet treeSet = new TreeSet(new Comparator() {
        @Override
        public int compare(Object o1, Object o2) {
        
            //下面 调用String的 compareTo方法进行字符串大小比较
            //return ((String) o2).compareTo((String) o1);
            
            //如果老韩要求加入的元素，按照长度大小排序
            return ((String) o1).length() - ((String) o2).length();
        }
    });