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方法就可以了.