区间加法（LeetCode）

题目

假设你有一个长度为 n 的数组，初始情况下所有的数字均为 0 ，你将会被给出 k ​​​​​​​ 个更新的操作。

其中，每个操作会被表示为一个三元组：[startIndex, endIndex, inc] ，你需要将子数组 A[startIndex ... endIndex] （包括 startIndex 和 endIndex）增加 inc。

请你返回 k 次操作后的数组。

解题

"""
这个问题可以使用差分数组来解决。
差分数组的思想是，通过记录差分，可以在常数时间内对一个区间的所有元素进行修改。
"""


def getModifiedArray(length, updates):
    # 初始化差分数组
    diff = [0] * (length + 1)

    # 处理每个操作
    for start, end, inc in updates:
        diff[start] += inc
        if end + 1 < length:
            diff[end + 1] -= inc

    # 根据差分数组计算最终数组
    result = [0] * length
    result[0] = diff[0]
    for i in range(1, length):
        result[i] = result[i - 1] + diff[i]

    return result


length = 5
updates = [
    [1, 3, 2],
    [2, 4, 3],
    [0, 2, -2]
]
print(getModifiedArray(length, updates))        # [-2, 0, 3, 5, 3]