假设我有以下课程:
class Z {
X x;
Y y;
}
我有一个Z元素列表。我想在一次传递中计算在x字段中有多少元素的值为x1,在y字段中有多少元素的值为y1。
使用循环是直截了当的:
int countOfx1 = 0;
int countOfy1 = 0;
for (Z z: list) {
if (z.x == x1) {
countOfx1++
}
if (z.y == y1) {
countOfy1++
}
}
可以简单地使用流吗?
答案 0 :(得分:5)
您可以通过为总计创建收集器来执行此操作:
class Zcount {
private int xCount = 0;
private int yCount = 0;
public Zcount accept(Z z) {
if (z.x == x1)
xCount++;
if (z.y == y1)
yCount++;
return this;
}
public Zcount combine(ZCount other) {
xCount += other.xCount;
yCount += other.yCount;
return this;
}
}
Zcount count = list.stream().collect(Zcount::new, Zcount::accept, Zcount::combine);
这比迭代解决方案更有优势,您可以使流并行,如果列表非常大,则可以提供性能优势。
答案 1 :(得分:2)
您可以使用我在this answer中发布的multiClassify
收藏家:
List<Predicates> preds = Arrays.asList(z -> z.x == x1, z -> z.y == y1);
List<Long> counts = stream.collect(multiClassify(preds, Collectors.counting()));
// counts.get(0) -> counts for z.x == x1
// counts.get(1) -> counts for z.y == y1
当然,简单的替代方法是遍历输入两次:
long countsX = list.stream().filter(z -> z.x == x1).count();
long countsY = list.stream().filter(z -> z.y == y1).count();
此类解决方案很短,通常在ArrayList
等常规输入的性能方面也不差。