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());
}
}
}