Java 8：map.merge 时间复杂度

Question

我试图找出下面代码的复杂性，因为 for 循环它将是 O(n * complexity_of_map.merge)

public int solution(int K, int[] A) {    
    Map<Integer, Integer> map = new HashMap<Integer, Integer>();
    for(int i =0; i < A.length; i++){
        map.merge(K - A[i], 1, Integer::sum);
    }   
    return Arrays.stream(A).map(element -> map.getOrDefault(element,0)).sum();
} 

谁能帮我理解上面代码的时间复杂度和map.merge() 在 Java 8 中。

Answer 1

引用自 JDK 8 的 Javadoc： https://docs.oracle.com/javase/8/docs/api/java/util/Map.html#merge-K-V-java.util.function.BiFunction-

默认实现等价于执行以下操作 此地图的步骤，然后返回当前值或 null if 缺席：

V oldValue = map.get(key);
V newValue = (oldValue == null) ? value :
          remappingFunction.apply(oldValue, value);
if (newValue == null)
    map.remove(key);
else
    map.put(key, newValue);

所有put、remove 和get 都是O(1) 的HashMap。您使用的remappingFunction 是Integer::sum，与n 无关。因此，您的解决方案中的 for 循环就是 O(n)。

对于流操作，stream + map + sum 应该大致相当于一个简单的for循环，这使得它成为O(n)。您传递给map() 的lambda 调用map.getOrDefault，这也是O(1) 的HashMap。所以这也是O(n) 的整体。

所以你的解决方案就是O(n)。