Java中的 TreeSet headSet() 方法及示例

Java中的TreeSet是一种基于红黑树的数据结构，具有元素自动排序的特性。TreeSet类中有一个headSet()方法，可以返回一个小于给定元素的子集。

headSet()方法

headSet()方法的定义如下：

public SortedSet<E> headSet(E toElement)

headSet()方法会返回一个小于给定元素的子集，其中包括起始元素。返回的集合会自动排序，并且是只读的。如果给定元素在集合中不存在，则返回小于给定元素的最大元素的子集。

示例

import java.util.TreeSet;
import java.util.SortedSet;

public class TreeSetHeadSetExample {
   public static void main(String[] args) {
      TreeSet<Integer> numbers = new TreeSet<Integer>();
      numbers.add(10);
      numbers.add(15);
      numbers.add(20);
      numbers.add(25);
      numbers.add(30);
      numbers.add(35);
      SortedSet<Integer> numbersHeadSet = numbers.headSet(25);
      System.out.println("TreeSet: " + numbers);
      System.out.println("HeadSet: " + numbersHeadSet);
   }
}

上述代码会输出以下结果：

TreeSet: [10, 15, 20, 25, 30, 35]
HeadSet: [10, 15, 20]

在示例中，首先创建了一个TreeSet对象numbers并添加了一些元素。然后，使用headSet()方法返回小于25的子集，并将结果分配给SortedSet对象numbersHeadSet。最后，使用System.out.println()方法输出原始集合和子集的元素。

总结

headSet()方法是TreeSet类的一个强大的方法，它返回小于给定元素的子集，其中包括起始元素。此外，返回的集合是只读的，并且自动排序。使用headSet()方法可以轻松地操作TreeSet集合中的数据。