找到 N 个通过给定条件的第一个元素：求和给定的数字

Question

我有Pocket 实例的列表

List<Pocket> pockets;

Pocket 类如下所示：

public class Pocket {
    String name;
    Double amount;
}

口袋列表示例：

[
    {"pocket1", 280},
    {"pocket2", 320},
    {"pocket3", 100},
    {"pocket4", 125},
    {"pocket5", 150},
    {"pocket6", 175}
]

我需要从给定的口袋列表中找到n 的第一个口袋，它们一起可以给出N（例如650）。如果最后一个口袋的amount 比需要的大，这个口袋应该分成两个口袋，结果列表应该只包含一个部分，与之前的n - 1 口袋一起给出N。

例如，我需要按照源列表中的顺序列出口袋列表，其中包含650 的数量。请注意，3-rd 口袋被拆分，只有一半的 amount 在最后一个口袋中返回：

[
    {"pocket1", 280},
    {"pocket2", 320},
    {"pocket3", 50}
]

如何用Java Streams实现？

Answer 1

您可以使用flatMap 和Stream 来执行此操作。见下例：

import java.util.Arrays;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Stream;

public class Lambda {

    public static void main(String[] args) {
        List<Pocket> pockets = Arrays.asList(
            new Pocket("pocket1", 280D),
            new Pocket("pocket2", 320D),
            new Pocket("pocket3", 100D),
            new Pocket("pocket4", 50D));

        System.out.println("Filtered");
        pockets.stream()
            .flatMap(new SplitFunction(651))
            .forEach(System.out::println);

        System.out.println("Original");
        pockets.forEach(System.out::println);
    }
}


class SplitFunction implements Function<Pocket, Stream<Pocket>> {

private double max;

public SplitFunction(double max) {
    this.max = max;
}

@Override
public Stream<Pocket> apply(Pocket pocket) {
    if (Double.compare(max, pocket.amount) >= 0) {
        max -= pocket.amount;
        return Stream.of(pocket);
    } else if (Double.compare(max, 0.0) > 0) {
        Pocket lastPocket = new Pocket(pocket.name, max);
        max = 0;

        return Stream.of(lastPocket);
    }
    return Stream.empty();
}
}

上面的代码打印：

Filtered
Pocket{name='pocket1', amount=280.0}
Pocket{name='pocket2', amount=320.0}
Pocket{name='pocket3', amount=51.0}
Original
Pocket{name='pocket1', amount=280.0}
Pocket{name='pocket2', amount=320.0}
Pocket{name='pocket3', amount=100.0}
Pocket{name='pocket4', amount=50.0}