Leetcode算法题【中等】370.区间加法

一.原题如下：

二.解题思路

1. 使用差分数组，构建一个差分数组工具类，方便调用。
2. startIndex相当于开区间，在开区间加上值
3. endIndex相当于闭区间，在闭区间的下一个索引减去值
4. 最后通过差分数组返回结果数组。

差分数组工具类

具体步骤请查看每个关键函数的代码注释

• 差分数组一共有三个函数需要注意
• 构造函数：将原始数组转换为差分数组
• 操作函数：在查分数组的基础上进行增减操作
• 结果数组：将操作完毕的差分数组重新转换为需要的原始数组

public class DifferenceHelper {
    // 定义差分数组
    int[] diff;

    public DifferenceHelper(int[] nums) {
        // 差分数组大小等同于原始数组大小
        diff = new int[nums.length];
        // 首个值都相等
        diff[0] = nums[0];
        // 查分数组当前值 = 原始数组当前值 - 原始数组上一个值
        for (int i = 1; i < nums.length; i++) {
            diff[i] = nums[i] - nums[i - 1];
        }
    }

    /**
     * 操作函数
     */
    public void operate(int start, int end, int val) {
        // 开区间加上值
        diff[start] += val;
        // 如果闭区间超过数组索引，则无需处理
        if (end + 1 >= diff.length) {
            return;
        }
        // 闭区间减去值
        diff[end + 1] -= val;
    }

    /**
     * 结果函数
     */
    public int[] result() {
        // 结果数组大小等同于差分数组大小
        int[] res = new int[diff.length];
        // 首个值都相等
        res[0] = diff[0];
        // 结果数组当前值 = 结果数组上一个值 + 差分数组当前值
        for (int i = 1; i < diff.length; i++) {
            res[i] = res[i - 1] + diff[i];
        }
        return res;
    }
}

三.完整题解

具体步骤请查看每个关键步骤的代码注释

    public int[] getModifiedArray(int length, int[][] updates) {
        // 构造差分数组
        DifferenceHelper diff = new DifferenceHelper(new int[length]);

        // 循环操作
        for (int[] update : updates) {
            // 获取三元组的开区间、闭区间、操作值
            int start = update[0];
            int end = update[1];
            int val = update[2];
            // 差分数组执行操作
            diff.operate(start, end, val);
        }

        // 差分数组返回结果
        return diff.result();
    }