Java TreeSet如何查找特定元素

在Java中，TreeSet是一个基于红黑树实现的有序集合，它提供了对元素的快速查找、插入和删除操作。要在TreeSet中查找特定元素，可以使用以下方法：

1. 使用contains()方法：这是最简单的方法，只需调用TreeSet的contains()方法并传入要查找的元素。如果元素存在于TreeSet中，该方法将返回true，否则返回false。

import java.util.TreeSet;

public class Main {
    public static void main(String[] args) {
        TreeSet<Integer> treeSet = new TreeSet<>();
        treeSet.add(1);
        treeSet.add(2);
        treeSet.add(3);

        int elementToFind = 2;
        boolean isPresent = treeSet.contains(elementToFind);
        System.out.println("Element " + elementToFind + " is present in the TreeSet: " + isPresent);
    }
}

2. 使用lower()和higher()方法：如果你想找到小于或等于给定元素的最大元素，可以使用lower()方法；如果想找到大于或等于给定元素的最小元素，可以使用higher()方法。

import java.util.TreeSet;

public class Main {
    public static void main(String[] args) {
        TreeSet<Integer> treeSet = new TreeSet<>();
        treeSet.add(1);
        treeSet.add(2);
        treeSet.add(3);

        int elementToFind = 2;
        Integer lowerElement = treeSet.lower(elementToFind);
        Integer higherElement = treeSet.higher(elementToFind);
        System.out.println("Lower element than " + elementToFind + ": " + lowerElement);
        System.out.println("Higher element than " + elementToFind + ": " + higherElement);
    }
}

请注意，TreeSet中的元素必须实现Comparable接口，或者在创建TreeSet时提供一个自定义的Comparator。否则，你可能会在运行时遇到ClassCastException。