使用java 8流计算单次传递多个项目

时间:2016-08-05 00:29:44

标签: java java-stream

假设我有以下课程:

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++
    }
 }

可以简单地使用流吗?

2 个答案:

答案 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等常规输入的性能方面也不差。