以有效的方式从 Map 中删除多个键？

Question

我有一个带有大量键值对的Map<String,String>。现在我想从Map 中删除选定的键。以下代码显示了我为实现这一目标所做的工作。

Set keySet = new HashSet(); //I added keys to keySet which I want to remove. 

然后：

Iterator entriesIterator = keySet.iterator();
while (entriesIterator.hasNext()) {
   map.remove( entriesIterator.next().toString());
} 

这工作正常。更好的方法是什么？

Answer 1

假设您的集合包含您要删除的字符串，您可以使用the keySet method 和map.keySet().removeAll(keySet);。

keySet 返回此映射中包含的键的 Set 视图。集合由地图支持，因此对地图的更改会反映在集合中，反之亦然。

人为的例子：

Map<String, String> map = new HashMap<>();
map.put("a", "");
map.put("b", "");
map.put("c", "");

Set<String> set = new HashSet<> ();
set.add("a");
set.add("b");

map.keySet().removeAll(set);

System.out.println(map); //only contains "c"

Answer 2

为了完成，并且当您寻找实现此目标的方法时，Google 会将您带到这里：

map.entrySet().removeIf(entry -> /* decide what you want to remove here */ );

这并不假定您有一组预定义的键要删除，而是假定您有一个条件，在该条件下应该删除这些键。从问题来看，尚不清楚这些键是手动添加还是基于某种条件。在后一种情况下，这可能是更简洁的代码。

对于前一种情况，这个（未经测试的）代码也可以工作：

map.entrySet().removeIf(entry -> keySet.contains(entry.getKey()) );

但显然the answer provided by @assylias 在这种情况下要干净得多！

Answer 3

为了完整起见：

正如猜测的 java.util.AbstractSet#removeAll 确实迭代了所有条目，但有一个小技巧：它使用较小集合的迭代器：

if (size() <= collection.size()) {
    Iterator<?> it = iterator();
    while (it.hasNext()) {
        if (collection.contains(it.next())) {
            it.remove();
        }
    }
} else {
    Iterator<?> it = collection.iterator();
    while (it.hasNext()) {
        remove(it.next());
    }
}

Answer 4

使用 Java 流：

keySet.forEach(map::remove);