String类的compareTo是通过字母的顺序进行排序,所以我们要自定义comparator比较器:
import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;
class ComparatorByLength implements Comparator{ //按长度比较,继承Comparator接口
public int compare(Object o1,Object o2){
String s1=(String)o1;
String s2=(String)o2;
int temp=s1.length()-s2.length();
return temp==0?s1.compareTo(s2):temp; //compareTo是字符串自身具备的方法
}
}
public class TreeSetDemo{
public static void main(String[] args){
TreeSet ts=new TreeSet(new ComparatorByLength()); //通过构造函数
ts.add("aaaaa");
ts.add("zz");
ts.add("nbds");
ts.add("cba");
ts.add("abc");
Iterator it=ts.iterator();
while(it.hasNext()){
System.out.println(it.next());
}
}
}
运行结果: