java中TreeSet的介绍与实现

在Java中，TreeSet 默认情况下使用其元素的自然顺序进行排序，也就是说，元素必须实现 Comparable 接口。如果需要自定义排序或者某些不自然排序的场景，可以通过构造一个 TreeSet 时传入一个自定义的 Comparator 实现来实现。

以下是自定义排序的一些常见场景：

1. 根据对象的某个属性排序：比如，根据用户的年龄而不是用户名排序。
2. 降序排序：默认是升序排序，可以通过 Comparator 实现降序。
3. 复合排序：比如，首先按一个属性排序，如果相同则按另一个属性排序。

示例代码

以下是一些自定义排序的示例：

1. 根据对象的某个属性排序

假设有一个 User 类，我们按 age 属性排序：

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

class User {
    String name;
    int age;

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

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

public class CustomSortTreeSet {
    public static void main(String[] args) {
        TreeSet<User> treeSet = new TreeSet<>(new Comparator<User>() {
            @Override
            public int compare(User u1, User u2) {
                return Integer.compare(u1.age, u2.age);
            }
        });

        treeSet.add(new User("Alice", 30));
        treeSet.add(new User("Bob", 25));
        treeSet.add(new User("Charlie", 35));

        for (User user : treeSet) {
            System.out.println(user);
        }
        // 输出：User{age=25, name=Bob}, User{age=30, name=Alice}, User{age=35, name=Charlie}
    }
}

2. 降序排序

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

class User {
    String name;
    int age;

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

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

public class DescendingSortTreeSet {
    public static void main(String[] args) {
        TreeSet<User> treeSet = new TreeSet<>(new Comparator<User>() {
            @Override
            public int compare(User u1, User u2) {
                return Integer.compare(u2.age, u1.age);
            }
        });

        treeSet.add(new User("Alice", 30));
        treeSet.add(new User("Bob", 25));
        treeSet.add(new User("Charlie", 35));

        for (User user : treeSet) {
            System.out.println(user);
        }
        // 输出：User{age=35, name=Charlie}, User{age=30, name=Alice}, User{age=25, name=Bob}
    }
}

3. 复合排序

先按年龄排序，如果年龄相同，则按名字排序：

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

class User {
    String name;
    int age;

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

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

public class CompositeSortTreeSet {
    public static void main(String[] args) {
        TreeSet<User> treeSet = new TreeSet<>(new Comparator<User>() {
            @Override
            public int compare(User u1, User u2) {
                int ageCompare = Integer.compare(u1.age, u2.age);
                if (ageCompare != 0) {
                    return ageCompare;
                } else {
                    return u1.name.compareTo(u2.name);
                }
            }
        });

        treeSet.add(new User("Alice", 30));
        treeSet.add(new User("Bob", 25));
        treeSet.add(new User("Charlie", 30));
        treeSet.add(new User("Alice", 25));

        for (User user : treeSet) {
            System.out.println(user);
        }
        // 输出：User{age=25, name=Alice}, User{age=25, name=Bob}, User{age=30, name=Alice}, User{age=30, name=Charlie}
    }
}

总结

• 通过传入自定义的 Comparator，可以轻松地实现 TreeSet 的自定义排序。
• 可以按对象的某个属性排序，实现降序或复合排序。

这些示例展示了如何使用 TreeSet 进行自定义排序，可以根据具体需求选择合适的排序方式。