黑马程序员技术交流社区

标题: 教学视频TreeSetDemo代码示例，复习下 [打印本页]

---

作者: 海狮    时间: 2015-10-14 15:29
标题: 教学视频TreeSetDemo代码示例，复习下

1. //使用TreeSet集合，添加自定义类，需复写判断对象是否相同的方法（CompareTo（Object obj））
   
2. import         java.util.*;
   
3. class TreeSetDemo
   
4. {
   
5.         public static void main(String[] args)
   
6.         {
   
7.                 TreeSet ts=new TreeSet();
   
8.                 ts.add(new Student("张三",3));
   
9.                 ts.add(new Student("李四",4));
   
10.                 ts.add(new Student("张五",6));
    
11.                 ts.add(new Student("王六",6));       
    
12.                 Iterator it=ts.iterator();
    
13.                 while(it.hasNext())
    
14.                 {
    
15.                 Student stu=(Student) it.next();
    
16.                 System.out.println(stu.getName()+"=="+stu.getAge());
    
17.                 }
    
18.         }
    
19. 
    
20. }
    
21. class Student implements Comparable
    
22. {
    
23.         private String name;
    
24.         private int age;
    
25.         Student(String name,int age){
    
26.         this.name=name;
    
27.         this.age=age;
    
28.         }
    
29.         public int compareTo(Object ob){//这里复写一个TreeSet判断对象相同的方法
    
30.                 if(!(ob instanceof Student))
    
31.                         throw new RuntimeException("不是学生对象");
    
32.                 Student stu=(Student) ob;
    
33.                 if(stu.getAge()>age)
    
34.                         return 1;
    
35.                 if(stu.getAge()==age)
    
36.         //age为主要条件，当主要条件相同时，得看需要判断次要条件
    
37.                         return this.name.compareTo(stu.name);
    
38.                 return -1;
    
39.         }
    
40.         public String getName(){
    
41.                 return name;
    
42.         }
    
43.         public int getAge(){
    
44.                 return age;
    
45.         }
    
46. }

复制代码

---

欢迎光临 黑马程序员技术交流社区 (http://bbs.itheima.com/)

黑马程序员IT技术论坛 X3.2