我试图将多个排序列表合并到一个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实现会更好.