集合的选择
TreeSet
即有序的Set。TreeSet的底层就是TreeMap
解读
- 当我们使用无参构造器,创建TreeSet时,仍然是无序的
- 老师希望添加的元素,按照字符串大小来排序
- 使用TreeSet提供的一个构造器,可以传入一个比较器(匿名内部类),并且指定排序规则
TreeSet treeSet = new TreeSet(new Comparator() {
@Override
public int compare(object o1,0bject o2) {
//下面调用String的 compareTo方法进行字符串大小比较
return ((String) o1).compareTo((string) o2);
}
});
构造器的底层
- 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;
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);
}
如果要求加入的元素,按照长度大小排序。那么只需要修改返回语句就可以
return ((String) o1).length() - ((String)o2).length();
这样是长度由小到大排序的。但是这样的话,长度相同的字符串会加不进去。
TreeMap
因为TreeSet的底层是TreeMap。所以用法基本与TreeSet一致。
public TreeSet() {
this(new TreeMap<>());
}
源码解读
- 1.构造器把传入的实现了Comparator接口的匿名内部类(对象),传给给TreeMap的comparator
public TreeMap(comparator<? super K>comparator) {
this.comparator = comparator;
}
- 2.调用put方法
- 2.1第一次添加,把k-v封装到Entry对象,放入root
Entry<K, V> t = root;
if (t == null){
compare(key,key); // type (and possibly null) check
root = new Entry<>(key, value,null);
size = 1;
modCount++;
return null;
}
- 2.2以后添加
Comparator<? super K> cpr = comparator;
if (cpr != null){
do {//遍历所有的key,给当前key找到适当位置
parent = t;
pmp = cpr.compare(key, t.key);//动态绑定到我们的匿名内部类的compare
if (cmp< 0)
t = t.left;
else if (cmp > 0)
t = t.right;
else///如果遍历过程中,发现准备添加Key 和当前已有的Key相等,就不添加
return t.setValue(value);
}while (t != null);
}
Collections工具类
我把Collecttions和Arrays两个工具类写到一个博客中了。
Collections和Arrays工具类