12 { return true; }
13 else { return false; }
14 }
15 《一线大厂Java面试题解析+后端开发学习笔记+最新架构讲解视频+实战项目源码讲义》无偿开源 威信搜索公众号【编程进阶路】 //通过重写compareTo方法,判断是否能加入Set里
16 public int compareTo(Object obj) {
17 // 判断是否是学生类型
18 if (obj instanceof Student) {
19 Student s = (Student) obj;
20 // 如果是学生类型,如果学号相等,则不加入Set
21 if (s.getId() == this.getId()) {
22 return 0;
23 } else {
24 return s.getId()>this.getId()?1:-1;
25 }
26 // 不是学生类型对象的话就不要加入它
27 } else { return 0; }
28 }
29 }
从第3到29行,我们定义了一个Student类,为了把它放入Set,我们需要实现Comparable接口,并重写其中的compareTo方法。
从第9到14行,我们重写了用于判断两个Student对象是否相等的equals方法,如果不写,将用Object的方法(这个方法是通过判断两个对象的地址是否一致来判断两个对象是否相等)。
TreeSet对象不会根据equals方法判断是否重复,也就是说,即使我们注释掉了这个方法,不会影响到运行结果,但在自定义类里,重写equals方法是个很好的习惯。
在第16行到28行的compareTo方法里,我们将根据返回int类型的值,执行对应的动作。
如果返回0,则表示该对象已经存在于Set里了,这个对象无法再次加入。
如果返回1或-1,则表示Set里还没有和它相同的对象,可以加入。具体的1和-1的差别将在后继讲述“Collections排序”时详细分析。
30 public class SetDupDemo {
31 public static void main(String[] args) {
32 Set intSet = new HashSet();
33 intSet.add(1);
34 intSet.add(1);
35 System.out.println(intSet.size()); //输出结果是1
36 Student s1 = new Student(1);
37 Student s2 = new Student(1);
38 Set stuSet = new TreeSet();
39 stuSet.add(s1);
40 stuSet.add(s2);
41 System.out.println(stuSet.size()); //输出结果是1
42 }
43 }
在main函数的第32行里,我们定义了一个HashSet对象,当我们在第33和34行放入两个相等的数值进去时,由于Set对象不允许重复值插入,所以其实只放入了一个,这点可以从第35行打印结果里得到验证。
在第38行,我们通过泛型的方式定义了一个只能容纳Student类的TreeSet对象,并在第39和40行放入了两个id都是1的Student对象。