如何实现Java中cumsum的具体操作步骤

原创

mob649e8168f1bb 2023-07-05 11:03:19 ©著作权

文章标签 数组 Java System 文章分类 Java 后端开发

©著作权归作者所有：来自51CTO博客作者mob649e8168f1bb的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java中cumsum

在Java中，cumsum是一种常用的操作，用于计算数组或集合中元素的累积和。cumsum是cumulative sum的缩写，意为累积和。

累积和的定义

累积和是指将一组数中的每个元素依次累加的结果。例如，对于数组[1, 2, 3, 4]，它的累积和为[1, 3, 6, 10]，即第一个元素不变，第二个元素为前两个元素之和，第三个元素为前三个元素之和，以此类推。

使用循环计算累积和

在Java中，可以使用循环来计算累积和。以下是一个使用for循环计算累积和的示例代码：

public class CumSumExample {
    public static void main(String[] args) {
        int[] array = {1, 2, 3, 4};
        int[] cumSum = new int[array.length];
        
        cumSum[0] = array[0];
        for (int i = 1; i < array.length; i++) {
            cumSum[i] = cumSum[i - 1] + array[i];
        }
        
        System.out.println("Cumulative sum: " + Arrays.toString(cumSum));
    }
}

在上述代码中，我们创建了一个数组array，并且定义了一个与array长度相同的数组cumSum来存储累积和。首先，我们将cumSum的第一个元素设为array的第一个元素，然后使用for循环从第二个元素开始计算累积和。循环中的cumSum[i] = cumSum[i - 1] + array[i]表示将cumSum的上一个元素与array的当前元素相加得到当前元素的累积和。最后，我们使用System.out.println打印出累积和数组。

运行以上代码，将输出Cumulative sum: [1, 3, 6, 10]。

使用Java 8的Stream计算累积和

在Java 8中，我们可以使用Stream API来更简洁地计算累积和。以下是使用Stream计算累积和的示例代码：

import java.util.Arrays;

public class CumSumExample {
    public static void main(String[] args) {
        int[] array = {1, 2, 3, 4};
        
        int[] cumSum = Arrays.stream(array)
                            .reduce((a, b) -> a + b)
                            .orElse(0);
        
        System.out.println("Cumulative sum: " + Arrays.toString(cumSum));
    }
}

在上述代码中，我们使用Arrays.stream(array)将数组转换为一个流，然后使用reduce方法来计算累积和。reduce方法接受一个BinaryOperator参数，它定义了如何将流中的元素进行累积操作。在本例中，我们使用lambda表达式(a, b) -> a + b将两个元素相加得到累积和。最后，我们使用orElse方法指定当数组为空时的默认值，这里设为0。最终，我们使用System.out.println打印出累积和数组。

运行以上代码，将输出Cumulative sum: [1, 3, 6, 10]。