TreeSet判断重复元素解析及代码示例
TreeSet的底层是TreeMap的keySet()，而TreeMap是基于红黑树实现的，红黑树是一种平衡二叉查找树，它能保证任何一个节点的左右子树的高度差不会超过较矮的那棵的一倍。
TreeMap是按key排序的，所以TreeSet中的元素也是排好序的。显然元素在插入TreeSet时compareTo()方法要被调用，所以TreeSet中的元素要实现Comparable接口。TreeSet作为一种Set，它不允许出现重复元素。TreeSet是用compareTo()来判断重复元素的，而非equals()，看下面代码。
importjava.util.TreeSet;
importorg.junit.Test;
publicclassTestTreeSet{
classCombineimplementsComparable{
privateintp1;
privateintp2;
publicCombine(intp1,intp2){
this.p1=p1;
this.p2=p2;
}
@Override
publicinthashCode(){
returnp1*31+p2;
}
@Override
publicBooleanequals(Objectobj){
System.out.print("whetherequal"+this+"and"+obj);
Booleanrect=false;
if(objinstanceofCombine){
System.out.println("whetherequal"+this+"and"+obj);
Combineother=(Combine)obj;
rect=(this.p1==other.getP1()&&this.p2==other.getP2());
}
System.out.println(":"+rect);
returnrect;
}
@Override
publicintcompareTo(Combineo){
System.out.print("compare"+this+"and"+o);
//排序时只考虑p1
if(this.p1o.p1){
System.out.println(",return1");
return1;
}else{
System.out.println(",return0");
return0;
}
}
@Override
publicStringtoString(){
return"("+p1+","+p2+")";
}
publicintgetP1(){
returnp1;
}
publicvoidsetP1(intp1){
this.p1=p1;
}
publicintgetP2(){
returnp2;
}
publicvoidsetP2(intp2){
this.p2=p2;
}
}
@Test
publicvoidtest(){
Combinec1=newCombine(1,2);
Combinec2=newCombine(1,2);
Combinec3=newCombine(1,3);
Combinec4=newCombine(5,2);
TreeSetset=newTreeSet();
set.add(c1);
set.add(c2);
set.add(c3);
set.add(c4);
while(!set.isEmpty()){
//按顺序输出TreeSet中的元素
Combinecombine=set.pollFirst();
System.out.println(combine.getP1()+"\t"+combine.getP2());
}
}
}
输出：
compare(1,2)and(1,2),return0
compare(1,2)and(1,2),return0
compare(1,3)and(1,2),return0
compare(5,2)and(1,2),return1
12
52
我们看到不论compareTo()返回的是不是相等，equals()方法都没有被调用。
总结
以上就是本文关于TreeSet判断重复元素解析及代码示例的全部内容，希望对大家有所帮助。感兴趣的朋友可以继续参阅本站其他相关专题，如有不足之处，欢迎留言指出。感谢朋友们对本站的支持！