使用java流从整数列表计算累积和的列表

vhipe2zx  于 2021-07-09  发布在  Java
关注(0)|答案(3)|浏览(404)

我有以下清单:

INPUT :: 4 5 8 -11 9 5 -7 4 6 -6 -8 -11 80 -32 -56 -15 5 -49 

 OUTPUT :: 4 9 17 6 15 20 13 17 23 17 9 -2 78 46 -10 -25 -20 -69

我需要计算累计和表的意义

T(n) = T(n) + T(n-1) for n >0; 
and
T(0) = T(0)

我想用javastreamapi来计算,这样我就可以用spark来实现大数据计算了。我对java流很幼稚,我尝试过几个表达式,但都不起作用等效的结构化代码应该是这样的:

int[] numbers = {4, 5, 8, -11, 9, 5, -7, 4, 6,-6, -8, -11, 80, -32, -56, -15, 5, -49};
int temp = 0;

for (int i = 0 ; i < numbers.length ; i++) {
   temp = temp + numbers[i];
   numbers[i] = temp;
}
p5fdfcr1

p5fdfcr11#

您可以尝试使用自定义收集器。

public static void main(String[] args) {
       List<Integer> integers = Arrays.asList(1, 2, 3, 4, 5);
       List<Integer> cumulatives = integers.stream().collect(CumulativeAdd.collector());
}
     private static final class CumulativeAdd {

       List<Integer> retArray= new ArrayList<>();
       int sum = 0; 

       public void accept(Integer num) {
           sum +=num;
           retArray.add(sum);
       }

       public CumulativeAdd combine(CumulativeAdd other) {
           throw new UnsupportedOperationException("Parallel Stream not supported");
       }

       public List<Integer> finish() {
           return retArray;
       }

       public static Collector<Integer, ?, List<Integer>> collector() {
           return Collector.of(CumulativeAdd::new, CumulativeAdd::accept, CumulativeAdd::combine, CumulativeAdd::finish);
       }

   }
uhry853o

uhry853o2#

这里有两种方法。
第一种方法效率很低,因为它基本上使用嵌套循环来累积值。第一个 IntStream 指定值的范围,嵌套的intstream创建一个变量范围,并对从0到该范围末尾的值求和。

int[] result1 = IntStream.range(0, vals.length).map(
        i -> IntStream.rangeClosed(0, i).map(k->vals[k]).reduce(0, (a, b) -> a + b))
        .toArray();

这个方法更符合传统的方法。对单个0数组进行流式处理,然后使用该数组累积值的连续和。

int[] result2 = Stream.of(new int[] { 0 })
        .flatMapToInt(k -> IntStream.of(vals).map(v -> {
            k[0] += v;
            return k[0];
        })).toArray();

System.out.println(Arrays.toString(result1));
System.out.println(Arrays.toString(result2));

两种打印

[4, 9, 17, 6, 15, 20, 13, 17, 23, 17, 9, -2, 78, 46, -10, -25, -20, -69]
[4, 9, 17, 6, 15, 20, 13, 17, 23, 17, 9, -2, 78, 46, -10, -25, -20, -69]

但你不能做得比这更好。

for (int i = 1; i < vals.length; i++) {
      vals[i] += vals[i-1];
}

底线是坚持你所拥有的。

wh6knrhe

wh6knrhe3#

试试这个。

int[] a = {4, 5, 8, -11, 9, 5, -7, 4, 6, -6, -8, -11, 80, -32, -56, -15, 5, -49};
Arrays.parallelPrefix(a, (x, y) -> x + y);
System.out.println(Arrays.toString(a));

输出:

[4, 9, 17, 6, 15, 20, 13, 17, 23, 17, 9, -2, 78, 46, -10, -25, -20, -69]

相关问题