黑马程序员———TreeSet两种排序方式

------Java培训、Android培训、iOS培训、.Net培训、期待与您交流！ -------

TreeSet对元素进行排序的方式一： 

让元素自身具备比较功能，元素就需要实现Comparable接口，覆盖compareTo方法。
示例：

1. import java.util.Comparator;
   
2. import java.util.Iterator;
   
3. import java.util.TreeSet;
   
4. 
   
5. //创建了一个根据Person类的name进行排序的比较器。
   
6. class ComparatorByName implements Comparator{
   
7. public int compare(Object o1,Object o2){
   
8. Person p1 = (Person)o1;
   
9. Person p2 = (Person)o2;
   
10. 
    
11. int temp = p1.getName().compareTo(p2.getName());
    
12. return temp == 0?p1.getAge()-p2.getAge() : temp;
    
13. }
    
14. }
    
15. 
    
16. public class TreeSetDemo{
    
17. public static void main(String[] args){
    
18. TreeSet ts = new TreeSet(new ComparatorByName());
    
19. 
    
20. //以Person对象年龄进行从小到大的排序
    
21. ts.add( new Person("zhangsan" ,28));
    
22. ts.add( new Person("wangwu" ,23));
    
23. ts.add( new Person("lisi" ,21));
    
24. ts.add( new Person("zhouqi" ,29));
    
25. ts.add( new Person("zhaoliu" ,25));
    
26. Iterator it = ts.iterator();
    
27. while(it.hasNext()){
    
28. Person p = (Person)it.next();
    
29. System.out.println(p.getName() + ":" + p.getAge());
    
30. }
    
31. }
    
32. }
    

复制代码

TreeSet集合第二种排序方式：
让集合自身具备比较功能，定义一个类实现Comparator接口，覆盖compare方法。将该类对象作为参数传递给TreeSet集合的构造函数。
示例2：

1. import java.util.Comparator;
   
2. import java.util.Iterator;
   
3. import java.util.TreeSet;
   
4. public class TreeSetTest{
   
5. public static void main(String[] args){
   
6. TreeSet ts = new TreeSet(new ComparatorByLen());
   
7. ts.add( "aaaa");
   
8. ts.add( "zz");
   
9. ts.add( "nbag");
   
10. ts.add( "cba");
    
11. ts.add( "abc");
    
12. Iterator it = ts.iterator();
    
13. while(it.hasNext()){
    
14.         System.out.println(it.next());
    
15. }
    
16. }
    
17. }
    
18. class ComparatorByLen implements Comparator{
    
19. public int compare(Object o1,Object o2){
    
20. String s1 = (String)o1;
    
21. String s2 = (String)o2;
    
22. int temp = s1.length() - s2.length();
    
23. return temp == 0?s1.compareTo(s2):temp;
    
24. }
    
25. }