【Leetcode】370. Range Addition

题目地址：

https://leetcode.com/problems/range-addition/

给定一个正整数$n$，想象有一个长$n$的数字全为零的数组$A$。再给出若干次update，每次update会把$[a,b]$区间内的数字都加一个数$x$。问经过$k$次update后数组变为什么。

思路是差分数组。开一个$d$数组，定义为$d[i]=A[i]-A[i-1]$。这样就有：$$A[k]=\sum _{i=0}^{k}d[i]$$也就是说，$A$和$d$两个数组可以互相得到，包含的信息是一样的。所以只需要对$d$做update，最后还原回来即可。而对$d$的update的时间是$O(1)$的。例如，update是将$A[a:b]$加上$x$，那么$A[a]-A[a-1]$就增大了$x$，也就是要将$d[a]+x$赋值给$d[a]$，并且$A[b+1]-A[b]$就减少了$x$，也就是要将$d[b+1]-x$赋值给$d[b+1]$（如果$b+1$出界了那就不用管），也就是每次update只需对$d$做两次操作即可。代码如下：

public class Solution {
    public int[] getModifiedArray(int length, int[][] updates) {
        int[] diff = new int[length];
        
        for (int[] update : updates) {
            diff[update[0]] += update[2];
            if (update[1] + 1 < length) {
                diff[update[1] + 1] -= update[2];
            }
        }
        
        for (int i = 1; i < length; i++) {
            diff[i] += diff[i - 1];
        }
        
        return diff;
    }
}

时间复杂度$O(n+u)$，$n$为数组长度，$u$为更新次数，空间$O(1)$。