TreeSet怎么用
时间: 2024-12-20 17:19:06 浏览: 4
TreeSet是Java集合框架中的一个有序集合,它实现了SortedSet接口,并内部使用红黑树数据结构。它的主要特点是自动排序,插入元素后会保持自然顺序或者自定义提供的比较器规定的顺序。
使用TreeSet的一般步骤如下:
1. **创建TreeSet**:
```java
TreeSet<String> treeSet = new TreeSet<>();
```
如果需要指定排序规则,可以传递一个Comparator实例给构造函数:
```java
Comparator<String> comparator = (a, b) -> a.compareTo(b);
TreeSet<String> treeSet = new TreeSet<>(comparator);
```
2. **添加元素**:
```java
treeSet.add("Apple");
treeSet.add("Banana");
treeSet.add("Cherry");
```
3. **查询、遍历和删除**:
- 查询特定元素是否存在:`treeSet.contains("Banana")`
- 遍历元素:`for (String fruit : treeSet) { System.out.println(fruit); }`
- 删除元素:`treeSet.remove("Apple");`
4. **迭代操作**:
```java
// 获取从最小到最大排序的所有元素
Iterator<String> iterator = treeSet.iterator();
while (iterator.hasNext()) {
String next = iterator.next();
// do something with the element
}
```
阅读全文