Java-TreeSet集合

1.1 集合体系

• Collection

  • List接口

    • ArrayList类

    • LinkedList类

  • Set接口

    • HashSet集合

    • TreeSet集合

1.2 TreeSet特点

• 不包含重复元素的集合[元素唯一]

• 没有带索引的方法[无索引]

• 可以按照指定的规则进行排序[可以排序]

• TreeSet 中的元素按照它们的自然顺序或者使用自定义的比较器进行排序。默认情况下，元素被排序为升序。
• 如果集合中的元素不是基本数据类型，那么它们必须实现 Comparable 接口来定义它们的自然排序顺序。
• 可以在创建 TreeSet 对象时提供一个实现了 Comparator 接口的比较器来定义元素的排序方式。例如：

  TreeSet<String> treeSet = new TreeSet<>(new MyComparator());

• 1.3treeset的简单练习

  //TreeSet的简单练习
  import java.util.TreeSet;
  
  public class TreeSetExample {
      public static void main(String[] args) {
          // 创建一个 TreeSet
          TreeSet<Integer> treeSet = new TreeSet<>();
  
          // 添加元素
          treeSet.add(5);
          treeSet.add(2);
          treeSet.add(8);
          treeSet.add(1);
  
          // 遍历输出元素
          for (Integer num : treeSet) {
              System.out.println(num);
          }
      }
  }
  

  1.4 排序规则

  1.4.1 自然排序

• 使用步骤

  • 使用空参构造创建TreeSet集合对象

  • 存储元素所在的类需要实现Comparable接口

  • 重写Comparable接口中的抽象方法 compareTo方法,指定排序规则

• compareTo方法如何指定排序规则 :

  • 此方法如果返回的是小于0 , 代表的是当前元素较小 , 需要存放在左边

  • 此方法如果返回的是大于0 , 代表的是当前元素较大, 需要存放在右边

  • 此方法如果返回的是0 , 代表的是当前元素在集合中已经存在 , 不存储

练习 : 存储学生对象, 按照年龄的升序排序，并遍历

//学生类定义
public class Student implements Comparable<Student> {
    private String name;
    private int age;

    // Constructor
    public Student(String name, int age) {
        this.name = name;
        this.age = age;
    }

    // Getter for age (assume getter for name is also defined)

    public int getAge() {
        return age;
    }

    // compareTo method
    @Override
    public int compareTo(Student o) {
        return this.age - o.age;
    }

    // toString method to display student information
    @Override
    public String toString() {
        return "Student{" +
               "name='" + name + '\'' +
               ", age=" + age +
               '}';
    }
}

//排序
import java.util.*;

public class Main {
    public static void main(String[] args) {
        // Create a TreeSet of Students
        TreeSet<Student> treeSet = new TreeSet<>();

        // Add some Student objects
        treeSet.add(new Student("Alice", 20));
        treeSet.add(new Student("Bob", 18));
        treeSet.add(new Student("Charlie", 22));
        treeSet.add(new Student("David", 19));

        // Print out the sorted Students
        for (Student student : treeSet) {
            System.out.println(student);
        }
    }
}

 

1.4.2 比较器排序

• 使用步骤

  • TreeSet的带参构造方法使用的是 “比较器排序” 对元素进行排序的

  • 比较器排序，就是让TreeSet集合构造方法接收Comparator接口的实现类对象

  • 重写Comparator接口中的 compare(T o1,T o2)方法 , 指定排序规则

  • 注意 : o1代表的是当前往集合中添加的元素 , o2代表的是集合中已经存在的元素,排序原理与自然排序相同!

• 排序规则

  • 排序原理与自然排序相同!

1.4.3 两种排序的区别

• 自然排序：自定义类实现Comparable接口，重写compareTo方法，根据返回值进行排序。

• 比较器排序：创建TreeSet对象的时候传递Comparator的实现类对象，重写compare方法，根据返回值进行排序。

• 如果Java提供好的类已经定义了自然排序排序规则 , 那么我们可以使用比较器排序进行替换

• 注意 : 如果自然排序和比较器排序都存在 , 那么会使用比较器排序

• 两种方式中，关于返回值的规则：

  • 如果返回值为负数，表示当前存入的元素是较小值，存左边

  • 如果返回值为0，表示当前存入的元素跟集合中元素重复了，不存

  • 如果返回值为正数，表示当前存入的元素是较大值，存右边