集合框架--TreeSet集合练习之字符串长度排序

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());
}
}
}运行结果：