TreeSet类的排序

一 TreeSet类的排序概述

TreeSet采用红黑树的数据结构对元素进行排序。TreeSet支持两种排序方法：自然排序和定制排序。

• 自然排序：TreeSet会调用集合元素的compareTo(Object obj)方法来比较元素之间大小关系，然后将集合元素按升序排列，这种方式就是自然排列。
• 定制排序：TreeSet借助于Comparator接口的帮助。该接口里包含一个的int compare(T o1, T o2)方法，该方法用于比较o1和o2的大小。

二 定制排序应用

1 代码示例

import java.util.*;class M{int age;public M(int age){this.age = age;}public String toString(){return "M[age:" + age + "]";}}public class TreeSetTest4{public static void main(String[] args){// 此处Lambda表达式的目标类型是ComparatorTreeSet ts = new TreeSet((o1 , o2) ->{M m1 = (M)o1;M m2 = (M)o2;// 根据M对象的age属性来决定大小，age越大，M对象反而越小return m1.age > m2.age ? -1: m1.age < m2.age ? 1 : 0;});ts.add(new M(5));ts.add(new M(-3));ts.add(new M(9));System.out.println(ts);}}

 

2 代码结果

[M[age:9], M[age:5], M[age:-3]]

3 代码说明

本代码在创建TreeSet集合对象时，提供一个Comparator对象与该TreeSet集合关联，由该Comparator对象负责集合元素的逻辑排序。由于Comparator是一个函数式接口，因此可使用Lambda表达式代替Comparator。