Java集合框架总结：TreeSet类的排序问题(1)(2)

2、定制排序

TreeSet的自然排序是根据集合元素的大小，TreeSet将他们以升序排列。如果需要实现定制排序，例如降序，则可以使用Comparator接口。该接口里包含一个int compareT o1， T o2）方法，该方法用于比较o1和o2的大小。

如果需要实现定制排序，则需要在创建TreeSet集合对象时，并提供一个Comparator对象与该TreeSet集合关联，由该Comparator对象负责集合元素的排序逻辑。

如下程序所示：

class M {  
int age;  
 
public M(int age) {  
this.age = age;  
}  
 
public String toString() {  
return "M对象age：" + age + ")";  
}  
}  
 
public class TestTreeSet3 {  
public static void main(String[] args) {  
TreeSet ts = new TreeSet(new Comparator() {  
public int compare(Object o1, Object o2) {  
 
M m1 = (M) o1;  
M m2 = (M) o2;  
 
if (m1.age > m2.age) {  
return -1;  
} else if (m1.age == m2.age) {  
return 0;  
} else {  
return 1;  
}  
}  
});  
ts.add(new M(5));  
ts.add(new M(-3));  
ts.add(new M(9));  
System.out.println(ts);  
}  
}  
 

程序运行结果：

[M对象age：9), M对象age：5), M对象age：-3)]

说明：

上面程序中创建了一个Comparator接口的匿名内部类对象，该对象负责ts集合的排序。所以当我们把M对象添加到ts集合中时，无须M类实现Comparable接口，因为此时TreeSet无须通过M对象来比较大小，而是由与TreeSet关联的Comparator对象来负责集合元素的排序。使用定制排序时，TreeSet对集合元素排序时不管集合元素本身的大小，而是由Comparator对象负责集合元素的排序规则。