1. import java.util.*;
   
2. /*
   
3. 需求：TreeSet集合两种存入元素的方法
   
4. */
   
5. class TreeSetTest
   
6. {
   
7.         public static void main(String[] args)
   
8.         {
   
9.                 TreeSet ts=new TreeSet(new Mycompare());
   
10.                 ts.add(new Student("张三",29));
    
11.                 ts.add(new Student("李四",29));
    
12.                 ts.add(new Student("王五",31));
    
13.                 ts.add(new Student("张三",29));
    
14.                 ts.add(new Student("赵六",30));
    
15.                 ts.add(new Student("李四",29));
    
16.                 sop(ts);
    
17. 
    
18.         }
    
19.         public static void sop(Object obj){
    
20.                 System.out.println(obj);
    
21.         }
    
22. }
    
23. //第一种：让需要存入 TreeSet集合的元素实现Comparable，从而具备比较性
    
24. class Student implements Comparable
    
25. {
    
26.         private String name;
    
27.         private int age;
    
28.         Student(String name,int age){
    
29.                 this.name=name;
    
30.                 this.age=age;
    
31.         }
    
32.         public String getName(){
    
33.                 return name;
    
34.         }
    
35.         public int getAge(){
    
36.                 return age;
    
37.         }
    
38.         //复写compareTo()方法，定义元素的比较方式
    
39.         public int compareTo(Object obj){
    
40.                 if(!(obj instanceof Student))
    
41.                         throw new RuntimeException("数据类型不符");
    
42.                 Student stu=(Student)obj;
    
43.                 int num=this.name.compareTo(stu.name);
    
44.                 if(num==0)
    
45.                         return new Integer(this.age).compareTo(stu.age);
    
46.                 return num;
    
47. 
    
48.         }
    
49.         //自定义equals方法，判断两个Person对象是否相同
    
50.         public boolean equals(Object obj){
    
51.                 if(!(obj instanceof Student))
    
52.                         return false;
    
53.                 Student stu=(Student)obj;
    
54.                 return this.getName().equals(stu.getName())&&this.age==stu.age;
    
55.         }
    
56.         public String toString(){
    
57.                 return this.name+":"+this.age;
    
58.         }
    
59. }
    
60. //第二种方式：自定义比较器，将比较器对象传给TreeSet集合对象，使集合具备对元素的比较功能
    
61. class Mycompare implements Comparator
    
62. {       
    
63.         public int compare(Object obj1,Object obj2){
    
64.                 Student stu1=(Student)obj1;
    
65.                 Student stu2=(Student)obj2;
    
66.                 int num=new Integer(stu1.getAge()).compareTo(stu2.getAge());
    
67.                 if(num==0)
    
68.                         return stu1.getName().compareTo(stu2.getName());
    
69.                 return num;
    
70.         }
    
71. }

复制代码

1. import java.util.*;
   
2. /*
   
3. 需求：TreeSet集合两种存入元素的方法
   
4. */
   
5. class TreeSetTest
   
6. {
   
7.         public static void main(String[] args)
   
8.         {
   
9.                 TreeSet ts=new TreeSet(new Mycompare());
   
10.                 ts.add(new Student("张三",29));
    
11.                 ts.add(new Student("李四",29));
    
12.                 ts.add(new Student("王五",31));
    
13.                 ts.add(new Student("张三",29));
    
14.                 ts.add(new Student("赵六",30));
    
15.                 ts.add(new Student("李四",29));
    
16.                 sop(ts);
    
17. 
    
18.         }
    
19.         public static void sop(Object obj){
    
20.                 System.out.println(obj);
    
21.         }
    
22. }
    
23. //第一种：让需要存入 TreeSet集合的元素实现Comparable，从而具备比较性
    
24. class Student implements Comparable
    
25. {
    
26.         private String name;
    
27.         private int age;
    
28.         Student(String name,int age){
    
29.                 this.name=name;
    
30.                 this.age=age;
    
31.         }
    
32.         public String getName(){
    
33.                 return name;
    
34.         }
    
35.         public int getAge(){
    
36.                 return age;
    
37.         }
    
38.         //复写compareTo()方法，定义元素的比较方式
    
39.         public int compareTo(Object obj){
    
40.                 if(!(obj instanceof Student))
    
41.                         throw new RuntimeException("数据类型不符");
    
42.                 Student stu=(Student)obj;
    
43.                 int num=this.name.compareTo(stu.name);
    
44.                 if(num==0)
    
45.                         return new Integer(this.age).compareTo(stu.age);
    
46.                 return num;
    
47. 
    
48.         }
    
49.         //自定义equals方法，判断两个Person对象是否相同
    
50.         public boolean equals(Object obj){
    
51.                 if(!(obj instanceof Student))
    
52.                         return false;
    
53.                 Student stu=(Student)obj;
    
54.                 return this.getName().equals(stu.getName())&&this.age==stu.age;
    
55.         }
    
56.         public String toString(){
    
57.                 return this.name+":"+this.age;
    
58.         }
    
59. }
    
60. //第二种方式：自定义比较器，将比较器对象传给TreeSet集合对象，使集合具备对元素的比较功能
    
61. class Mycompare implements Comparator
    
62. {       
    
63.         public int compare(Object obj1,Object obj2){
    
64.                 Student stu1=(Student)obj1;
    
65.                 Student stu2=(Student)obj2;
    
66.                 int num=new Integer(stu1.getAge()).compareTo(stu2.getAge());
    
67.                 if(num==0)
    
68.                         return stu1.getName().compareTo(stu2.getName());
    
69.                 return num;
    
70.         }
    
71. }

复制代码