Java学习笔记 --- TreeSet和TreeMap

一、TreeSet源码解读

package com.javase.set_;

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

public class TreeSet01 {
    public static void main(String[] args) {
        //1. 当我们使用无参构造器，创建TreeSet时，仍然是无序的
        //2. 希望添加的元素，按照字符串大小来排序
        //3. 使用TreeSet 提供的一个构造器，可以传入一个比较器(匿名内部类) 并指定排序规则

        //TreeSet treeSet = new TreeSet();
        TreeSet treeSet = new TreeSet(new Comparator() {
            @Override
            public int compare(Object o1, Object o2) {
                //下面调用 String 的 compareTo 方法比较字符串大小
                return ((String) o1).compareTo((String) o2);
            }
        });
        //添加数据
        treeSet.add("jack");
        treeSet.add("tom");
        treeSet.add("sp");
        treeSet.add("a");

        System.out.println(treeSet);

        //源码解读
        /*
        1. 构造器把传入的比较器对象，赋给了 TreeSet的底层 TreeMap 的属性 this.comparator
           public TreeMap(Comparator<? super K> comparator) {
                  this.comparator = comparator;
           }
        2. 在 调用 TreeSet.add("tom")时 底层会执行到
           if (cpr != null) { //cpr 就是我们的匿名内部类(对象)
                do {
                    parent = t;
                    //动态绑定到我们的匿名内部类对象 comparator
                    cmp = cpr.compare(key, t.key);
                    if (cmp < 0)
                        t = t.left;
                    else if (cmp > 0)
                        t = t.right;
                    else //如果相等 即等于0 这个key就不加入
                        return t.setValue(value);
                } while (t != null);
            }
         */
    }
}

二、TreeMap源码解读

import java.util.Comparator;
import java.util.TreeMap;

public class TreeMap01 {
    public static void main(String[] args) {
        //1. 当我们使用无参构造器，创建TreeMap时，仍然是无序的
        //2. 希望添加的元素，按照字符串大小来排序
        //3. 使用TreeMap 提供的一个构造器，可以传入一个比较器(匿名内部类) 并指定排序规则
        //TreeMap treeMap = new TreeMap();
        TreeMap treeMap = new TreeMap(new Comparator() {
            @Override
            public int compare(Object o1, Object o2) {
                return ((String) o1).compareTo((String) o2);
            }
        });

        treeMap.put("jack", "张三");
        treeMap.put("tom", "李四");
        treeMap.put("smith", "老六");
        treeMap.put("kristina", "铁蛋");

        System.out.println(treeMap);

        //源码解读
        /*
        1. 构造器 把传入的比较器对象，赋给了 TreeMap 的属性 this.comparator
           public TreeMap(Comparator<? super K> comparator) {
                  this.comparator = comparator;
           }
        2. 在 调用 put 方法时
           (1)第一次添加，把k-v封装到Entry对象，放入root
           (2)后续添加时 底层会执行到
           if (cpr != null) { //cpr 就是我们的匿名内部类(对象)
                do { //遍历所有key 给当前的key找到适当的位置
                    parent = t;
                    //动态绑定到我们的匿名内部类对象 comparator
                    cmp = cpr.compare(key, t.key);
                    if (cmp < 0)
                        t = t.left;
                    else if (cmp > 0)
                        t = t.right;
                    else //如果遍历过程中 key相等 即等于0 这个key就不加入
                        return t.setValue(value);
                } while (t != null);
            }
         */
    }
}