首先,力扣第 370 题「区间加法」 就直接考察了差分数组技巧:

假设你有一个长度为 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]

那么我们直接复用刚才实现的 Difference 类就能把这道题解决掉:

int[] getModifiedArray(int length, int[][] updates) {
    // nums 初始化为全 0
    int[] nums = new int[length];
    // 构造差分解法
    Difference df = new Difference(nums);
    
    for (int[] update : updates) {
        int i = update[0];
        int j = update[1];
        int val = update[2];
        df.increment(i, j, val);
    }
    
    return df.result();
}