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

复制代码

使用泛型写这样的代码将会更好