TreeSet自定义排重

TreeSet集合通常是用来对某个对象进行自定义规则排序,下面我们举个例子说明一下:

public static void main(String[] args) {
		TreeSet<persion> a = new TreeSet<>();
		Collections.addAll(a, new persion("小明", 17), new persion("小王", 17), new persion("小李", 19),
				new persion("小刘", 10));
		System.out.println(a);
	}
}


public class persion implements Comparable<persion>{
  String name;
  int age;
  public persion() {}
public persion(String name, int age) {
	super();
	this.name = name;
	this.age = age;
}
@Override
public String toString() {
	return "persion [name=" + name + ", age=" + age + "]";
}
@Override
public int compareTo(persion per) {
	
    if(this.age>per.age){
        return 1 ; //升序
    }else if(this.age<per.age){
        return -1 ; //降序
    }else{
        return 0 ;    
    }
}

打印结果为:

[persion [name=小刘, age=10], persion [name=小明, age=17], persion [name=小李, age=19]]

从结果中我们可以发现小王因为年龄与小明相同,从而被系统排重了.

如果我们只想要进行姓名排重,仅仅修改下persion类中的compareTo方法即可

public int compareTo(persion per) {
	
    if(this.age>per.age){
        return 1 ;
    }else if(this.age<per.age){
        return -1 ;
    }else{
        return this.name.compareTo(per.name) ;    
    }
}

打印结果为:

[persion [name=小刘, age=10], persion [name=小明, age=17], persion [name=小王, age=17], persion [name=小李, age=19]]

与HashSet相比较,Hashset仅仅重写Hashcode与equals方法就可以了.