java-如何使用二进制搜索从排序的TreeSet中检索元素?

我试图将多个排序列表合并到一个TreeSet中.然后,我考虑在该TreeSet上应用Binary Search算法,以O(log n)时间复杂度来检索元素.

下面是我的代码,在其中我以一种方法传递列表列表,并将它们组合到TreeSet中,以避免重复…输入中的所有列表均已排序-

private TreeSet<Integer> tree = new TreeSet<Integer>();

public void mergeMultipleLists(final List<List<Integer>> inputs) {
    tree = new TreeSet<Integer>();
    for (List<Integer> input : inputs) {
        for(Integer ii : input) {
            tree.add(ii);
        }
    }
}

public List<Integer> getItem(final Integer x) {
    // extract elements from TreeSet in O(log n)
}

>首先,这是将多个排序列表合并到TreeSet中的正确方法吗?是否有任何直接方法可以有效地合并TreeSet中的多个排序列表?
>其次,我将如何以O(log n)时间复杂度从该TreeSet中提取元素?我想在该TreeSet中找到一个元素x,如果存在,则返回它,如果不存在,则从TreeSet返回下一个最大值.

还是与当前使用的数据结构相比,我可以使用其他数据结构?

更新的代码:

私有TreeSet树= new TreeSet();

public SearchItem(final List<List<Integer>> inputs) {
    tree = new TreeSet<Integer>();
    for (List<Integer> input : inputs) {
        tree.addAll(input);
    }
}

public Integer getItem(final Integer x) {
    if(tree.contains(x)) {
        return x;
    } else {
        // now how do I extract next largest 
         // element from it if x is not present
    }
}

解决方法:

TreeSet由NavigableMap(特别是TreeMap)支持.在TreeSet上调用contains()代表TreeMap.containsKey(),这是一个二进制搜索实现.

您可以使用TreeSet.contains()检查对象是否包含在集合中,但是必须首先拥有该对象.如果您希望能够查找和检索对象,则Map实现会更好.

上一篇:Unbiased Teacher for Semi-Supervised Object Detection论文解读


下一篇:Java:首次添加调用时触发的TreeSet compareTo方法