TreeSet はオブジェクトをどのように並べ替えますか

TreeSet は Comparator を渡すことでオブジェクトを並べ替えることができます。TreeSet のデフォルト コンストラクターを使用してインスタンスが作成される場合、要素は自然な順序付けを提供するために Comparable インターフェイスを実装する必要があります。要素が Comparable インターフェイスを実装していない場合、TreeSet インスタンスの作成時に Comparator オブジェクトを渡すことで並べ替えを実行できます。

たとえば、Student クラスがあるとします。

public class Student {
    
    
    private int id;
    private String name;
    private int age;

    // constructor, getters, and setters omitted for brevity

    @Override
    public String toString() {
    
    
        return "Student{" +
                "id=" + id +
                ", name='" + name + '\'' +
                ", age=" + age +
                '}';
    }
}

年齢によって昇順に並べ替えるコンパレーターを実装できます。

import java.util.Comparator;

public class AgeComparator implements Comparator<Student> {
    
    
    @Override
    public int compare(Student s1, Student s2) {
    
    
        return s1.getAge() - s2.getAge();
    }
}

次に、TreeSet インスタンスを作成するときに AgeComparator を渡して、年齢の昇順で並べ替えます。

import java.util.TreeSet;

public class TreeSetExample {
    
    
    public static void main(String[] args) {
    
    
        TreeSet<Student> studentSet = new TreeSet<>(new AgeComparator());
        studentSet.add(new Student(1, "Alice", 18));
        studentSet.add(new Student(2, "Bob", 20));
        studentSet.add(new Student(3, "Charlie", 19));
        System.out.println(studentSet);
    }
}

出力は次のとおりです。

[Student{id=1, name='Alice', age=18}, Student{id=3, name='Charlie', age=19}, Student{id=2, name='Bob', age=20}]