流中的地图元素对应于帖子
我经常发现自己在做这样的事情:
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
类的图书馆可以完成同样的工作,或者你可以创建自己的图书馆。
上一篇: map element in stream corresponding to post
下一篇: Order of growth as worst case running time as a function of N