TreeSet comparator字符串长度排序

TreeSet comparator字符串长度排序

package com.Set;

import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;

/**
 * TreeSet 存储字符串   根据字符串长度排序
 * @author 小明
 *
 */
class StringLengthComparator implements Comparator{

	@Override
	public int compare(Object o1, Object o2) {
		String s1 = (String) o1;
		String s2 = (String) o2;
		int num = new Integer(s1.length()).compareTo(new Integer(s2.length()));
		if(num ==0){
			return s1.compareTo(s2);//字符串排序
		}
		return num;
	}
	
}
public class TreeSetString {
public static void main(String[] args) {
	TreeSet<String> ts = new TreeSet<String>(new StringLengthComparator());
	ts.add("aa");
	ts.add("aac");
	ts.add("aacd");
	ts.add("aadd");
	ts.add("d");
	Iterator<String> it = ts.iterator();
	while(it.hasNext()){
		System.out.println(it.next());
	}
		
}
}