ITEEDU

入门准备

接触对象

对象导向

资源管理

对象容器(Container)

输入输出(I/O)

执行绪(Thread)

反射(Reflection)

metadata

数据库(JDBC)

舍遗补缺

Java Gossip: TreeSet

TreeSet实作Set接口与SortedSet接口,提供相关的方法让您有序的取出对应位置的对象,像是 first()、last()等方法,TreeSet是J2SE中唯一实作SortedSet接口的类别,它使用红黑树结构来对加入的对象进行排序。

看个简单的例子:

TreeSetDemo.java
package onlyfun.caterpillar;
import java.util.*;
public class TreeSetDemo {
	public static void main(String[] args) {
		Set set = new TreeSet();
		set.add("justin");
		set.add("caterpillar");
		set.add("momor");
		set.add("justin");
		Iterator iterator = set.iterator();
		while(iterator.hasNext()) {
			System.out.print(iterator.next() + " ");
		}
		System.out.println();
	}
}

由于加入的是String对象,执行结果会自动依字典顺序进行排序的动作:

caterpillar  justin momor

依字典顺序排序String对象是TreeSet预设的,如果您对对象有自己的一套排序顺序,您要实作一个 Comparator 对象,您要实作compare()方法,它必须传回整数值,如果对象顺序相同则传回0,传回正整数表示compare()方法的第一个对象大于第二个对象,反之则传回负整数。

举个实际的例子,假设您想要改变TreeSet依字典顺序排列加入的对象为相反的顺序:

CustomComparator.java
package onlyfun.caterpillar;
import java.util.Comparator;
public class CustomComparator implements Comparator {
	public int compare(T o1, T o2) {
		if (((T) o1).equals(o2))
		return 0;
		return ((Comparable) o1).compareTo((T) o2) * -1;
	}
}

在自订的Comparator中,如果两个对象的顺序相同会传回0,这在TreeSet中表示两个对象是同一个对象,TreeSet要求传入的对象必须实 作java.lang.Comparable接口,范例中只是简单的将原来compareTo()传回的值乘以负一,如此在TreeSet中就可以简单的 让排列顺序相反。

在建构TreeSet实例时一并指定自订的Comparator,例如:

TreeSetDemo2.java
package onlyfun.caterpillar;
import java.util.*;
public class TreeSetDemo2 {
	public static void main(String[] args) {
		// 自订Comparator
		Comparator comparator =
		new CustomComparator();
		Set set =
		new TreeSet(comparator);
		set.add("justin");
		set.add("caterpillar");
		set.add("momor");
		// 使用 enhanced for loop 显示对象
		for(String name : set) {
			System.out.print(name + " ");
		}
		System.out.println();
	}
}

执行的结果是相反的:

momor justin  caterpillar