我想知道,如果流(或收集器)中已经实现了一个已将列表排序为值的功能。例如。以下代码均按年龄分类,生成按性别分组的人员名单。第一个解决方案有一些开销排序(看起来有点邋))。第二种解决方案需要两次看每个人,但是要以一种漂亮的方式完成工作。
首先排序然后在一个流中分组:
apply plugin: 'findbugs'
findbugs {
toolVersion = '3.0.1'
effort = 'max'
reportLevel = 'medium'
}
首先分组,然后对每个值进行排序:
Map<Gender, List<Person>> sortedListsByGender = (List<Person>) roster
.stream()
.sorted(Person::compareByAge)
.collect(Collectors.groupingBy(Person::getGender));
我只是想知道,如果已经实施了某些内容,可以在一次运行中执行此操作,例如Map<Gender, List<Person>> sortedListsByGender = (List<Person>) roster
.stream()
.collect(Collectors.groupingBy(Person::getGender));
sortedListsByGender.values()
.forEach(list -> Collections.sort(list, Person::compareByAge));
。
答案 0 :(得分:19)
在sorted(comparator)
操作之前在流上使用collect
时,流必须缓冲整个流内容以便能够对其进行排序,并且排序可能涉及该缓冲区内的更多数据移动,与之后对小组的较小列表进行排序相比较。因此,如果已启用并行处理,则实现将使用多个内核,性能不如对各个组进行排序。
但请注意,使用sortedListsByGender.values().forEach(…)
不是可并行化的操作,即使使用sortedListsByGender.values().parallelStream().forEach(…)
也只允许并行处理组,而每个排序操作仍然是顺序的。
在收集器中执行排序操作时,如
static <T> Collector<T,?,List<T>> toSortedList(Comparator<? super T> c) {
return Collectors.collectingAndThen(
Collectors.toCollection(ArrayList::new), l->{ l.sort(c); return l; } );
}
Map<Gender, List<Person>> sortedListsByGender = roster.stream()
.collect(Collectors.groupingBy(Person::getGender, toSortedList(Person::compareByAge)));
排序操作的行为相同(感谢Tagir Valeev纠正我),但您可以轻松检查插入排序策略的执行方式。只需将收集器实现更改为:
static <T> Collector<T,?,List<T>> toSortedList(Comparator<? super T> c) {
return Collectors.collectingAndThen(
Collectors.toCollection(()->new TreeSet<>(c)), ArrayList::new);
}
为了完整性,如果你想要一个首先插入分类到ArrayList
的收集器以避免最后的复制步骤,你可以使用更精细的收集器,如下所示:
static <T> Collector<T,?,List<T>> toSortedList(Comparator<? super T> c) {
return Collector.of(ArrayList::new,
(l,t) -> {
int ix=Collections.binarySearch(l, t, c);
l.add(ix<0? ~ix: ix, t);
},
(list1,list2) -> {
final int s1=list1.size();
if(list1.isEmpty()) return list2;
if(!list2.isEmpty()) {
list1.addAll(list2);
if(c.compare(list1.get(s1-1), list2.get(0))>0)
list1.sort(c);
}
return list1;
});
}
它对顺序使用有效,但其合并功能不是最佳的。基础排序算法将受益于预先排序的范围,但必须首先找到这些范围,尽管我们的合并函数实际上知道这些范围。不幸的是,JRE中没有公共API允许我们使用这些信息(有效;我们可以将subList
传递给binarySearch
,但为list2
的每个元素创建一个新的子列表可能会变成出去太贵了)。如果我们想进一步提高并行执行的性能,我们必须重新实现排序算法的合并部分:
static <T> Collector<T,?,List<T>> toSortedList(Comparator<? super T> c) {
return Collector.of(ArrayList::new,
(l,t) -> l.add(insertPos(l, 0, l.size(), t, c), t),
(list1,list2) -> merge(list1, list2, c));
}
static <T> List<T> merge(List<T> list1, List<T> list2, Comparator<? super T> c) {
if(list1.isEmpty()) return list2;
for(int ix1=0, ix2=0, num1=list1.size(), num2=list2.size(); ix2<num2; ix2++, num1++) {
final T element = list2.get(ix2);
ix1=insertPos(list1, ix1, num1, element, c);
list1.add(ix1, element);
if(ix1==num1) {
while(++ix2<num2) list1.add(list2.get(ix2));
return list1;
}
}
return list1;
}
static <T> int insertPos(
List<? extends T> list, int low, int high, T t, Comparator<? super T> c) {
high--;
while(low <= high) {
int mid = (low+high)>>>1, cmp = c.compare(list.get(mid), t);
if(cmp < 0) low = mid + 1;
else if(cmp > 0) high = mid - 1;
else {
mid++;
while(mid<=high && c.compare(list.get(mid), t)==0) mid++;
return mid;
}
}
return low;
}
请注意,与简单的基于binarySearch
的插入不同,此最后一个解决方案是一个稳定的排序实现,即在您的情况下,Person
具有相同的年龄且Gender
不会如果源流具有已定义的遭遇顺序,则更改其相对顺序。