JavaのTreeSetでのカスタムソート方法を教えてください。

2年 ago
優斗, 朝陽
1 minute

Comparatorインターフェースを実装することで、独自の並べ替えを実装することができます。このComparatorインターフェースには、compareメソッドがあり、独自の並べ替え規則に基づいて2つのオブジェクトの大小を比較することができます。

以下は、Personクラスをカスタマイズし、年齢でTreeSetにソートする例です。

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

class Person {
    private String name;
    private int age;

    public Person(String name, int age) {
        this.name = name;
        this.age = age;
    }

    public String getName() {
        return name;
    }

    public int getAge() {
        return age;
    }
}

class AgeComparator implements Comparator<Person> {
    @Override
    public int compare(Person p1, Person p2) {
        return p1.getAge() - p2.getAge();
    }
}

public class Main {
    public static void main(String[] args) {
        TreeSet<Person> set = new TreeSet<>(new AgeComparator());
        set.add(new Person("Alice", 25));
        set.add(new Person("Bob", 30));
        set.add(new Person("Charlie", 20));

        for (Person person : set) {
            System.out.println(person.getName() + " - " + person.getAge());
        }
    }
}

結果は出力されます。

Charlie - 20
Alice - 25
Bob - 30

この例では、AgeComparatorクラスがComparatorインターフェースを実装し、compareメソッドをオーバーライドして、Personオブジェクトのageプロパティに基づいて比較するようにしています。次に、TreeSetオブジェクトを作成する際に、AgeComparatorオブジェクトをパラメータとして渡すことで、TreeSetはAgeComparatorで定義されたソートルールに従って並べ替えを行います。

bannerAds