目录
TreeSet的继承关系
- 由上图可以看出TreeSet继承了AbstractSet抽象类,实现了NavigableSet接口。
- TreeSet的本质是一个有序的,并且没有重复元素的集合,它是通过TreeMap实现的。TreeSet中含有一个"NavigableMap类型的成员变量"m,而m实际上是TreeMap的实例。
TreeSet任何判断重复元素
- 由于TreeSet是通过TreeMap实现的,它添加的元素实际上就是TreeMap的Key值
- 而TreeMap在判断是否有相同Key值时,是使用comparator来比较两个对象是否相等的,而非equal()方法比较。
- 因此在向TreeSet中添加元素时,即使两个元素对象的Hash值不同,但是通过Comparator比较后结果为相同,依然会添加失败。
- 现在我们来创建一个TreeSet,里面储存的元素类型为数组,这个数组包含两个值,我们让第一个值均不一样,而第二个值均为0,而comparator比较的是第二个值的大小,我们尝试一下添加多个元素,查看结果。
public class Test {
public static void main(String[] args) {
TreeSet<int[]> treeSet = new TreeSet<>(new Comparator<int[]>() {
@Override
public int compare(int[] o1, int[] o2) {
return o1[1]-o2[1];
} //比较第二个值的大小
});
for (int i = 0; i < 10; i++) {
treeSet.add(new int[]{i,0});
} //第一个值均不一样(0~10递增),第二个值均为0
for (int[] ints : treeSet) {
System.out.println(Arrays.toString(ints));
} //打印出set中所有的元素
}
}
- 从打印结果中我们可以看出,我们预期添加10个元素到TreeSet中,而10个元素的第一个值都不一样,如果是以hash值判断的话,这些元素都互不相同,而最终只有一个元素添加成功,这就说明其他元素都被当做重复元素处理了
改写Comparator添加重复元素以及出现的问题
- 通过以上分析,我们可以尝试通过修改Comparator,在其结果为0时,强制返回一个非零的数,来达到添加重复元素的目的
TreeMap<Integer,String> tree = new TreeMap<Integer,String>(new Comparator<Integer>(){
@Override
public int compare(Integer o1, Integer o2) {
if(o1.equals(o2))
return 1;
else
return o1-o2;
}
});
-
问题来了,比如我们在compare中重写,判断两个元素一样是,强制返回1,元素是存进去了,可是我们在调用get方法去拿出这个元素和其值时,就出问题了,拿到的永远是null,也就是找不到这个元素!不按常规出牌总是要付出代价的,为什么呢??查看底层代码,我们可以发现,get方法取元素同样是根据比较器来取,调用compare方法查找,直到查找到能让compare方法返回0的元素,就判断,这个元素就是我们要找的元素,就将它的值取出来,如果查完了也没有返回0,那么就认为集合中不存在这个键,返回null!
-
虽然用这种方法可以将重复的元素储存进去,当时相应的问题也会出现,当我们想使用ceiling(E e)或floor(E e)方法获得大于等于e的最大值,或小于等于e的最小值的时候,由于相等时返回值被强制修改为非0的值,所以无法获得等于e的值。
-
同样的道理,TreeSet其实就是通过TreeMap来实现的,那么其也响应的特性。(比如ceiling()和floor()等等)