leetcode —— 区间加法

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

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

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

示例:

输入: length = 5, updates = [[1,3,2],[2,4,3],[0,2,-2]]
输出: [-2,0,3,5,3]

解释：

初始状态:
[0,0,0,0,0]
进行了操作 [1,3,2] 后的状态:
[0,2,2,2,0]
进行了操作 [2,4,3] 后的状态:
[0,2,5,5,3]
进行了操作 [0,2,-2] 后的状态:
[-2,0,3,5,3]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/range-addition
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

——————————————

class Solution:
    def getModifiedArray(self, length: int, updates: List[List[int]]) -> List[int]:
        ans = [0 for _ in range(length+1)]   # 用于保存每个操作的信息
        for up in updates:
            begin,end,num = up[0],up[1],up[2]
            
            # 对每个操作的起始位置和结束位置进行记录
            ans[begin] += num  
            ans[end+1] -= num
            
        temp = ans[0]  # temp用于保存ans[0-n]的总操作数累加和
        for i in range(1,length):
            if ans[i] != 0:
                temp += ans[i]
                
            ans[i] = temp
        return ans[:-1]

时间复杂度为O(n+k)，空间复杂度为O(1)；