从集合中删除“第一个”对象

Question

在某些情况下，我需要驱逐 Java Set 中最旧的元素。该集合是使用LinkedHashSet 实现的，这很简单：只需去掉集合迭代器返回的第一个元素即可：

Set<Foo> mySet = new LinkedHashSet<Foo>();
// do stuff...
if (mySet.size() >= MAX_SET_SIZE)
{
    Iterator<Foo> iter = mySet.iterator();
    iter.next();
    iter.remove();
}

这很难看：如果我使用 SortedSet，我可以用 1 行代码完成 3 行（由于其他原因，SortedSet 不是这里的选项）：

if (/*stuff*/)
{
    mySet.remove(mySet.first());
}

那么有没有更简洁的方法来做到这一点，没有：

• 更改Set 实现，或
• 编写静态实用方法？

任何利用Guava 的解决方案都可以。

---

_{我完全知道集合没有内在的顺序。我问的是删除迭代顺序定义的第一个条目。}

Answer 1

LinkedHashSet 是 LinkedHashMap 的包装器，它支持简单的“删除最旧”策略。要将其用作 Set，您可以这样做

Set<String> set = Collections.newSetFromMap(new LinkedHashMap<String, Boolean>(){
    protected boolean removeEldestEntry(Map.Entry<String, Boolean> eldest) {
        return size() > MAX_ENTRIES;
    }
});

Answer 2

if (!mySet.isEmpty())
  mySet.remove(mySet.iterator.next());

似乎不到 3 行。

如果您的集合由多个线程共享，您当然必须围绕它进行同步。

Answer 3

如果您确实需要在代码中的多个位置执行此操作，只需编写一个静态方法即可。

提出的其他解决方案通常速度较慢，因为它们暗示调用Set.remove(Object) 方法而不是Iterator.remove() 方法。

@Nullable
public static <T> T removeFirst(Collection<? extends T> c) {
  Iterator<? extends T> it = c.iterator();
  if (!it.hasNext()) { return null; }
  T removed = it.next();
  it.remove();
  return removed;
}

Answer 4

用番石榴：

if (!set.isEmpty() && set.size() >= MAX_SET_SIZE) {
    set.remove(Iterables.get(set, 0));
}

我还会建议另一种方法。是的，它改变了实现，但不会彻底改变：扩展 LinkedHashSet 并在 add 方法中具有该条件：

public LimitedLinkedHashSet<E> extends LinkedHashSet<E> {
    public void add(E element) {
         super.add(element);
         // your 5-line logic from above or my solution with guava
    }
}

它仍然是 5 行，但它对使用它的代码是不可见的。而且由于这实际上是集合的特定行为，因此将其包含在集合中是合乎逻辑的。

Answer 5

我认为你这样做的方式很好。这是您经常做的事情，是否值得寻找更短的方法？你可以像这样用 Guava 做同样的事情：

Iterables.removeIf(Iterables.limit(mySet, 1), Predicates.alwaysTrue());

这增加了包装集合及其迭代器以进行限制然后调用alwaysTrue() 谓词一次的小开销……但对我来说似乎并不特别值得。

编辑：要将我在评论中所说的话放在答案中，您可以创建一个SetMultimap，它会自动限制每个键可以拥有的值的数量，如下所示：

SetMultimap<K, V> multimap = Multimaps.newSetMultimap(map,
    new Supplier<Set<V>>() {
      public Set<V> get() {
        return Sets.newSetFromMap(new LinkedHashMap<V, Boolean>() {
          @Override protected boolean removeEldestEntry(Entry<K, V> eldestEntry) {
            return size() > MAX_SIZE;
          }
        });
      }
    });

Answer 6

又快又脏的单线解决方案：mySet.remove(mySet.toArray(new Foo[mySet.size()])[0]) ;)

但是，我仍然会选择迭代器解决方案，因为它更具可读性并且应该更快。

编辑：我会选择 Mike Samuel 的解决方案。 :)