集合框架--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());
		}
	}
}

运行结果：