使用java8流接口累加查找限制为10的索引

Question

我正在学习java8 stream api。我不得不面对这个问题。这里有9个数字"1,2,3,4,5,6,7,8,9“。我想为1+2+3+....+n>=10找到n。如何使用java8流接口查找n？

我尝试使用for循环来查找n，代码如下：

int sum,n=0;
for(int i=0;i<arr.length;i++){
    sum+=arr[i]
    if(sum>=10){
        n=i;
        break;
    }
}

Answer 1

int n = IntStream.range(0, arr.length)
        .filter(i -> Arrays.stream(arr).limit(i + 1).sum() >= 10)
        .findFirst()
        .orElse(-1);
System.out.println(n); // return 3 (1 + 2 + 3 + 4 = 10)

Answer 2

您的代码与您的问题描述相矛盾。您说您想要n这样的1+2+3+....+n>=10，但是您将数组中n的索引i赋值给n。只有当数组中有这些升序数字时，这两个才相关，但如果假设内容是固定的，则根本不需要该数组。

此外，您根本没有递增sum；它一直是零。

如果您只假设一个升序整数序列，则无需迭代即可获得n：

int threshold = 10;

long n = Math.round(Math.sqrt(threshold*2));//that's it
System.out.println("solution: n = "+n);

// the rest is only verification and reporting
long actualSum = LongStream.rangeClosed(1, n).sum();
assert actualSum >= threshold;
assert actualSum - n < threshold;

System.out.println(LongStream.rangeClosed(1, n)
    .mapToObj(String::valueOf).collect(Collectors.joining(" + ")));
System.out.println("\t = "+actualSum+" >= "+threshold);

如果想要得到存储在数组中的任意序列的解决方案，循环是最简单、最有效的顺序解决方案：

int i, sum;
for(i = 0, sum = 0; i < arr.length && sum < threshold; i++) sum += arr[i];

if(sum >= threshold) {
    System.out.println("solution: index = "+(i-1)+", n = "+arr[i-1]);
    System.out.println(Arrays.stream(arr, 0, i)
            .mapToObj(String::valueOf).collect(Collectors.joining(" + ")));
    System.out.println("\t = "+sum+" >= "+threshold);
} else {
    System.out.println("No solution with this array");
}

如果您有一个大型数组，并且希望利用并行处理，则可以使用

Arrays.parallelPrefix(arr, Integer::sum);
int i = Arrays.binarySearch(arr, threshold);
if(i < 0) i = -i-1;

if(i < arr.length){
    System.out.println("solution: index = "+i+", n = "+(arr[i]-(i==0? 0: arr[i-1])));
    System.out.println(arr[0]+" + "+IntStream.rangeClosed(1, i)
        .mapToObj(ix -> String.valueOf(arr[ix]-arr[ix-1]))
        .collect(Collectors.joining(" + ")));
    System.out.println("\t = "+arr[i]+" >= "+threshold);
} else {
    System.out.println("No solution with this array");
}

但这会修改数组。这种修改是可逆的，如示例所示，我们可以打印原始序列，但这样的修改并不总是可行的。如果我们必须复制或将数组转换回其原始状态，我们可能已经失去了并行处理的任何优势，因此循环在这些情况下仍然是最佳选择。您无论如何都需要一个足够大的数组，才有机会从并行处理中获得性能优势。