Java中TreeSet添加重复元素问题

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()等等）