Data Structure & Algorithms
DSA
Arrays prefix sum
In Place Prefix Sum

In-place Prefix Sum

Problem Description

Given an array A of N integers. Construct prefix sum of the array in the given array itself.

Problem Constraints

1 <= N <= 10^5
1 <= A[i] <= 10^3

Input Format

Only argument A is an array of integers.

Output Format

Return an array of integers denoting the prefix sum of the given array.

Example Input

Input 1:
A = [1, 2, 3, 4, 5]

Input 2:
A = [4, 3, 2]

Example Output

Output 1:
[1, 3, 6, 10, 15]

Output 2:
[4, 7, 9]

Example Explanation

Explanation 1:
The prefix sum array of [1, 2, 3, 4, 5] is [1, 3, 6, 10, 15].

Explanation 2:
The prefix sum array of [4, 3, 2] is [4, 7, 9].

Output

Java
public int[] prefixSum(int[] A) {
    int n = A.length;
    for (int i = 1; i < n; i++) {
        A[i] += A[i - 1];
    }
    return A;
}
Python
def prefix_sum(A):
    for i in range(1, len(A)):
        A[i] += A[i - 1]
    return A
JavaScript
function prefixSum(A) {
    for (let i = 1; i < A.length; i++) {
        A[i] += A[i - 1];
    }
    return A;
}