假设我在数据结构中有一些整数。当我在数据结构中插入新数字时, 我希望获得新插入元素与数据结构中已有的任何其他元素之间的最小差异。我应该使用什么数据结构和算法? O(n)解决方案是微不足道的,我想要更好。 感谢。
答案 0 :(得分:5)
一种选择是使用TreeSet
(基于TreeMap
),这需要多次O(lg n)
次操作。该类公开了两种方法,可用于查找最接近您要插入的值的元素:
公共E上限(E e)
返回此set中大于或等于给定元素的最小元素,如果没有这样的元素,则返回null。公众E楼(E e)
返回此set中小于或等于给定元素的最大元素,如果没有这样的元素,则返回null。
public static int findClosest(TreeSet set, Integer val) {
if (set == null || set.size() == 0) {
return -1;
}
// ceiling == 9 for input of 7
// O(lg n) operation
Integer ceiling = (Integer)set.ceiling(val);
// floor = 6 for input of 7
// O(lg n) operation
Integer floor = (Integer)set.floor(val);
if (ceiling == null) {
return val - floor;
}
if (floor == null) {
return ceiling - val;
}
return (val - floor > ceiling - val) ? ceiling - val : val - floor;
}
public static void main(String[] args) {
TreeSet<Integer> ts = new TreeSet<>();
ts.add(5);
ts.add(1);
ts.add(6);
ts.add(9);
ts.add(2);
ts.add(3);
int diff = findClosest(ts, 7);
// closest is 6, so diff == 1
}
答案 1 :(得分:0)
您可以使用height-balanced binary search tree。这些可以用许多数据结构之一来实现。插入通常平均为O(log(n)),查找一个或两个最接近的整数(在插入值的两侧)也最多为O(log(n))。
可能有其他数据结构可以做得更好(特别是如果你可以合理地绑定你需要处理的整数值),但我想不出一个我的头脑。