Skip to Content

JavaTreeSet和TreeMap

TreeSet

TreeSet实现了Set接口,是一个有序的集合。它继承了AbstractSet抽象类,实现了NavigableSet<E>Cloneable,Serializable接口。TreeSet是基于TreeMap实现的,TreeSet的元素支持2种排序方式:自然排序或者根据提供的Comparator进行排序

使用方式和Set集合差不多

自然排序是默认的,也可以使用Comparator进行排序:

TreeSet treeSet = new TreeSet(new Comparator() { @Override public int compare(Object o1, Object o2) { // 自定义按照字符串的顺序升序排序 return o1.toString().compareTo(o2.toString()); } }); treeSet.add("zhangsan"); treeSet.add("lisi"); treeSet.add("arbor"); treeSet.add("wangwu");

① 调用的是带有比较器的构造方法,可以看到在内部,new了一个TreeMap的对象,并把比较器传给了TreeMap

② TreeMap中有一个comparator属性,会将默认的比较器替换为传入的比较器。

③ 调用add方法,可以看到,底层调用的是put方法,实际上也是TreeMap的put方法

④ 而在put方法中:

重点: 如果比较器得到的结果为0,是不添加该元素,而是替换value的值,因为TreeSet没有value,所以直接不添加元素。 . 比如,按照两个字符串的长度排序,如果字符串长度一样,就添加不进去。

public V put(K key, V value) { Entry<K,V> t = root; // 这里省略一些代码,主要作用是: // 第一次添加元素时,也会触发比较器,自己和自己比较 // 主要是为了预防传入null值 int cmp; Entry<K,V> parent; // cpr就是自己传入的比较器 // 然后根据比较规则添加元素 Comparator<? super K> cpr = comparator; if (cpr != null) { do { parent = t; cmp = cpr.compare(key, t.key); if (cmp < 0) t = t.left; else if (cmp > 0) t = t.right; else // 如果比较的时候相等,cmp==0的情况下: // 这个数据就不加入(因为是set集合,单列的) // 如果不是从TreeSet过来的,会替换value值 return t.setValue(value); } while (t != null); } // 如果没有传比较器,就走这条线,按照默认规则排序 else { if (key == null) throw new NullPointerException(); @SuppressWarnings("unchecked") Comparable<? super K> k = (Comparable<? super K>) key; do { parent = t; cmp = k.compareTo(t.key); if (cmp < 0) t = t.left; else if (cmp > 0) t = t.right; else return t.setValue(value); } while (t != null); } Entry<K,V> e = new Entry<>(key, value, parent); if (cmp < 0) parent.left = e; else parent.right = e; fixAfterInsertion(e); size++; modCount++; return null; }

TreeMap

TreeMap的使用方式和Map接口使用方式差不多。

Java中的TreeMap用于存储与HashMap类非常相似的键值对。区别在于TreeMap提供了一种以排序顺序存储键/值对的有效方法。它是基于红黑树的NavigableMap实现。

key不能为null,但value可以为null。

TreeMap的元素支持2种排序方式:自然排序或者根据提供的Comparator进行排序

TreeMap treeMap = new TreeMap(new Comparator() { @Override public int compare(Object o1, Object o2) { return o1.toString().length() - o2.toString().length(); } }); treeMap.put("zhangsan", "张三"); treeMap.put("lisi", "李四"); treeMap.put("arbor", "乔木先生"); treeMap.put("wangwu", "王五"); treeMap.put("xiaoming", "小明");

执行过程和TreeSet一样,不同的是如果调用compare方法,得到的结果为0,则替换value值。 而TreeSet因为没有value,所以不会添加元素。

Last updated on