Chương trình tìm tổng tiền tố của mảng 1D trong Python

Cho mảng nums. Tổng tiền tố rs[i] là tổng các phần tử từ nums[0] đến nums[i]. Trả về mảng tổng tiền tố.

Ví dụ: nums = [8,3,6,2,1,4,5] → [8, 11, 17, 19, 20, 24, 29].

Thuật toán

• n = len(nums), rs = [nums[0]].
• Với i từ 1 đến n-1:
• nums[i] = nums[i] + nums[i-1]
• Thêm nums[i] vào rs.
• Trả về rs.

Ví dụ (Python)

def solve(nums):
    n = len(nums)
    rs = [nums[0]]
    for i in range(1, n):
        nums[i] += nums[i-1]
        rs.append(nums[i])
    return rs

nums = [8,3,6,2,1,4,5]
print(solve(nums))

Đầu vào

[8,3,6,2,1,4,5]

Đầu ra

[8, 11, 17, 19, 20, 24, 29]