java - 如何从两个映射中删除相同的键

Question

我在 Java 中创建了两个包含相同类型的键和值的映射:

Map<Position,Movement> A;
Map<Position,Movement> B;

我想从它们中删除(不是创建新 map )相同的键。我不在乎值(value)是否不同。例如，如果 A有职位: a2 , 机芯: n,n和 B有职位: a2 , 机芯: 1,2应该删除这些条目。

我想知道是否有一种快速的方法可以在不迭代最短 map 并比较每个键的情况下做到这一点。

谢谢

Answer 1

您可以使用 Iterator在 keySet()如果键存在于另一个映射中，则从迭代器和另一个映射中删除元素。

Iterator<String> itr = map1.keySet().iterator();
String key;
while (itr.hasNext()) {
    key = itr.next();
    if (map2.containsKey(key)) {
        itr.remove();
        map2.remove(key);
    }
}

演示:

import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

public class Main {
    public static void main(String[] args) {
        Map<String, String> map1 = new HashMap<String, String>();
        map1.put("a", "1");
        map1.put("b", "1");
        map1.put("x", "1");
        map1.put("c", "1");
        map1.put("z", "1");
        map1.put("d", "1");

        Map<String, String> map2 = new HashMap<String, String>();
        map2.put("w", "1");
        map2.put("x", "1");
        map2.put("b", "1");
        map2.put("y", "1");
        map2.put("c", "1");
        map2.put("z", "1");

        Iterator<String> itr = map1.keySet().iterator();
        String key;
        while (itr.hasNext()) {
            key = itr.next();
            if (map2.containsKey(key)) {
                itr.remove();
                map2.remove(key);
            }
        }

        System.out.println(map1);
        System.out.println(map2);
    }
}

输出:

{a=1, d=1}
{w=1, y=1}

[更新]

使用 Stream 的解决方案:

map1.keySet()
    .stream()
    .filter(k -> map2.containsKey(k))
    .collect(Collectors.toList())
    .forEach(k -> {
        map1.remove(k);
        map2.remove(k);
    });

[另一个更新]

下面给出的是第一个解决方案的紧凑版本(感谢 Holger ):

for (Iterator<String> itr = map1.keySet().iterator(); itr.hasNext();) {
    if (map2.keySet().remove(itr.next())) {
        itr.remove();
    }
}