TreeSet的使用

package test;

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

public class TreeSetDemo {

	public static void main(String[] args) {
		// TODO 自動生成的方法存根
		TreeSet t = new TreeSet(new MyComparator());
		t.add("abcdefs");
		t.add("bcd");
		t.add("fjakfj");
		t.add("hello");
		System.out.println(t);
	}
}

//定義自己的比較器，實現字符串按照字符串長度比較。
class MyComparator implements Comparator{

	@Override
	public int compare(Object arg0, Object arg1) {
		// TODO 自動生成的方法存根
		String s1 = (String) arg0;
		String s2 = (String) arg1;
		int flag = s1.length() - s2.length();
		if(flag>0){
			return 1;
		}else if(flag == 0){
			return 0;
		}else
			return -1;
	}
	
}