Java集合TreeSet用法详解

1. 什么是TreeSet

TreeSet是Java集合框架中的一种实现，它是一个有序的、支持基本操作（添加、删除、查找）的集合。使用TreeSet可以方便地实现对元素的排序，并且支持非重复元素的存储。

在TreeSet中，元素按照自然顺序或者指定的比较器顺序进行排序，其中自然顺序指元素实现Comparable接口，并且根据compareTo方法返回的值进行比较。如果集合中元素没有实现Comparable接口，也可以在创建TreeSet时指定一个比较器Comparator。

2. TreeSet的常用方法

2.1 添加元素

TreeSet提供了两种添加元素的方法：

• add(Element e)：将指定的元素添加到集合中；
• addAll(Collection c)：将指定集合中的所有元素添加到集合中。

2.2 删除元素

TreeSet提供了三种删除元素的方法：

• clear()：从集合中删除所有元素；
• remove(Object o)：从集合中删除指定的元素；
• removeAll(Collection c)：从集合中删除指定集合中的所有元素。

2.3 查找元素

TreeSet提供了以下查找元素的方法：

• contains(Object o)：如果集合中包含指定元素，返回true；
• first()：返回集合中的第一个（最小）元素；
• last()：返回集合中的最后一个（最大）元素。

2.4 获取集合大小

TreeSet提供了以下获取集合大小的方法：

• size()：返回集合中元素的数量。

3. TreeSet的示例

3.1 使用自然顺序排序

下面是一个使用自然顺序进行排序的TreeSet示例，元素类型为String。

import java.util.TreeSet;

public class TreeSetExample {
    public static void main(String[] args) {
        // 创建TreeSet集合
        TreeSet<String> set = new TreeSet<>();
        // 添加元素
        set.add("apple");
        set.add("banana");
        set.add("orange");
        set.add("pear");
        // 遍历集合
        for (String s : set) {
            System.out.println(s);
        }
    }
}

输出结果如下：

apple
banana
orange
pear

3.2 使用比较器进行排序

下面是一个使用比较器进行排序的TreeSet示例，元素类型为Person，按照年龄从小到大进行排序。

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

public class TreeSetExample {
    public static void main(String[] args) {
        // 创建比较器
        Comparator<Person> comparator = new Comparator<Person>() {
            @Override
            public int compare(Person p1, Person p2) {
                return p1.getAge() - p2.getAge();
            }
        };
        // 创建TreeSet集合
        TreeSet<Person> set = new TreeSet<>(comparator);
        // 添加元素
        set.add(new Person("Tom", 25));
        set.add(new Person("Jerry", 20));
        set.add(new Person("Lucy", 30));
        // 遍历集合
        for (Person p : set) {
            System.out.println(p);
        }
    }
}

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 void setName(String name) {
        this.name = name;
    }

    public int getAge() {
        return age;
    }

    public void setAge(int age) {
        this.age = age;
    }

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

输出结果如下：

Person{name='Jerry', age=20}
Person{name='Tom', age=25}
Person{name='Lucy', age=30}

4. 总结

通过本文的介绍，我们了解了TreeSet的使用方法以及常用操作。在实际开发中，可以根据需要选择自然顺序排序或者指定比较器进行排序。