如何使用Binary Search从排序的TreeSet中检索元素？

Question

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

下面是我的代码，在其中，我以一种方法传递列表列表，并将它们组合在一起TreeSet以避免重复…里面的所有列表inputs都经过排序-

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中提取元素？我想找到的元素x在TreeSet，如果有，则返回它，如果它不存在则返回从下一个最大的价值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
    }
}

Answer 1

TreeSet由NavigableMapa
TreeMap专门支持。调用contains()的TreeSet委托TreeMap.containsKey()，这是一个二进制搜索实现。

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