【Leetcode 经典题目】370. Range Addition [Math][DP]

Assume you have an array of length n initialized with all 0's and are given k update operations.

Each operation is represented as a triplet: [startIndex, endIndex, inc] which increments each element of subarray A[startIndex ... endIndex] (startIndex and endIndex inclusive) with inc.

Return the modified array after all k operations were executed.

Example:

Input: length = 5, updates = [[1,3,2],[2,4,3],[0,2,-2]]
Output: [-2,0,3,5,3]

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

---

【PS】这是一道看上去很简单的题目，可以简单的遍历每个update操作，然后直接在对应数组位置进行加减。然而tricky的地方在于，提示里说时间复杂度可以是O(K+N), 并且不需要额外空间，也不需要复杂的数据结构。于是就花了一上午时间来思考这道题目的巧妙算法，如果是面试的话估计早就挂了（不建议这样的刷题方法，太没效率）。最后终于想出来这种时间复杂度可以beat 100%的算法，发现是一道数学证明题。先上一个自己手写的公式：

 

想明白这个算法以后，代码就很简单。

1. 遍历update，对数组首尾位置数字进行加减 （+[up][j"][2]-up[k"][2)的部分）

2.从i=1开始遍历数组，加上前一个元素即可。

【代码】

class Solution {
public:
    vector<int> getModifiedArray(int length, vector<vector<int>>& updates) {
        vector<int> res(length, 0);
        for(auto up: updates) {
            res[up[0]] += up[2];
            if(up[1] < length-1) res[up[1]+1] -= up[2];
        }
        for(int i = 1; i < length; ++i)
            res[i] += res[i-1];
        return res;
    }
};