流中的地图元素对应于帖子

我经常发现自己在做这样的事情:

list.stream().min(new Comparator<>() {

    @Override
    public int compare(E a, E b) {
        return Double.compare(f(a),f(b));
    }
})

其中f是一个计算密集型函数。 这需要对实际需要的f两倍的评估。 我更喜欢

list.stream().mapToDouble(f).min()

但后来我不知道如何获得这个最小值对应的原始元素。

一个丑陋的解决方法是

class WithF<E>{
    private final E e;
    private final double fe;
    WithF(E e, double fe){
        this.e = e;
        this.fe = fe;
    }
    public E getE(){
        return e;
    }
    public double getFE(){
        return fe;
    }
}

接着

list.stream().map(e -> new WithF<>(e,f(e))).min(Comparator.comparingDouble(WithF::getFE))

使用流式API做这件事情有没有更好的,习惯用法的方式?


这种转换通常称为Schwartzian变换

实际上,我会用一些额外的位来概括WithF ,并将其重命名为使管道整洁。

class SchwartzianKV<E, SORTKEY implements Comparable<SORTKEY> > 
      implements Comparable<SchwartzianKV<E, SORTKEY>> {

    public final E e;
    public final SORTKEY sortkey;

    SchwartzianKV(E e, SORTKEY sortkey){
        this.e = e;
        this.sortkey = sortkey;
    }

    public static <E, SORTKEY implements Comparable<SORTKEY>> 
    Function<E, SchwartzianKV<E, SORTKEY>> transformer( Function<E, SORTKEY> fn ) {
       return new Function<E, SchwartzianKV<E,SORTKEY>>() {
           @Override SchwartzianKV<E,SORTKEY> apply(E e) {
               return new SchwartzianKV<>(e, fn.apply(e));
           } 
       }
    }

    public int compare(With<E> other) {
       return sortkey.compare(other.sortkey);
    }
}

现在你可以写流

 Optional<E> minValue = list.stream()
                            .map( SchwartianKV.transformer( e -> f(e) ) )
                            .min()
                            .map( kv -> kv.e )

这很简洁。


在等待时我会发布我目前正在考虑的内容:

List<Double> fs = list.stream()
    .map(e -> f(e))
    .collect(Collectors.toList())
int i = IntStream.range(0,fs.size()).boxed()
    .min(java.util.Comparator.comparingDouble(fs::get))
    .get();
list.get(i)

怎么样:

Optional<E> minE = list.stream()
    .map(e -> new AbstractMap.SimpleEntry(e, f(e))
    .min(Map.Entry.comparingByValue())
    .map(Map.Entry::getKey);

这本质上是为每个列表项目创建一个映射条目到double值,然后使用这个值找到min条目,最后得到匹配键。

请注意,它返回一个可Optional以允许列表中没有项目的情况,在这种情况下,您将获得一个empty 。 或者,如果列表为空,您可以在末尾添加orElse调用以返回另一个E

这是Map.Entry稍微不寻常的用法(即不把它放在地图中)。 有一些Pair类的图书馆可以完成同样的工作,或者你可以创建自己的图书馆。

链接地址: http://www.djcxy.com/p/87635.html

上一篇: map element in stream corresponding to post

下一篇: Order of growth as worst case running time as a function of N