Java 8 哈希映射

Question

我有一张地图Map<String, List<Double>，我想在所有列表中找到最大值（或最小值）。 该函数应返回最大值（或最小值）和属于该值的键。

签名可能是

public static Pair<String,Double> getKeyValue(Map<String, List<Double>> map, BinaryOperator<Double> function)

获取地图和函数Double::max或Double::min

如何使用 java 8、stream api 高效（且美观）地实现这一点？

Answer 1

BinaryOperator 不是该任务的良好规范，它可以直接用于归约以产生适当的值，例如最小值或最大值，但它不适合返回关联值，例如 Map 的键值。以这种方式使用它意味着实现必须执行额外的操作来找出BinaryOperator 实际做了什么，以便在减少期间选择正确的键值。更糟糕的是，它不能保证BinaryOperator 做了一些允许执行这种减少的事情，例如运算符可能返回一个既不是它的参数的值。

对于这样的任务，Comparator 是更好的选择，因为它旨在指定排序并执行相关操作，例如查找最大值和最小值。实现可能如下所示：

public static Pair<String,Double> getMinimumKeyValue(
    Map<String, List<Double>> map, Comparator<Double> function) {

    return map.entrySet().stream()
        .map(e->new Pair<>(e.getKey(), e.getValue().stream().min(function).get()))
        .min(Comparator.comparing(Pair::getRight, function)).get();
}

它被命名为getMinimumKeyValue，因为它会在你传入Comparator.naturalOrder()时返回最小的键/值对。

但是你也可以通过Comparator.reverseOrder()获得最大值。

而且它很容易修改以支持更广泛的用例：

public static <K,V> Pair<K,V> getMinKeyValue(
    Map<K, ? extends Collection<V>> map, Comparator<? super V> function) {

    return map.entrySet().stream()
        .map(e->new Pair<>(e.getKey(), e.getValue().stream().min(function).get()))
        .min(Comparator.comparing(Pair::getRight, function)).get();
}

这仍然适用于从Map<String, List<Double>> 中获取Pair<String,Double>，但可以做更多事情......

Answer 2

你可以试试这个：

public static Pair<String,Double> getKeyValue(Map<String,List<Double>> map,
        BinaryOperator<Double> function) {
        return map.entrySet().stream()
            .map(e -> new Pair<String,Double>(e.getKey(),e.getValue().stream().reduce(function).get()))
            .reduce((p1,p2) -> function.apply(p1.getValue(),p2.getValue()).equals(p1.getValue()) ? p1 : p2)
            .get();
}

说明：

首先，将每个Entry<String,List<Double>> 关联到一个Pair<String,Double>，其中值是列表的最小值（或最大值）并且键保持不变，然后比较每个Pair<String,Double> 以找到具有最低（或最高）值。