LeetCode 370. 區間加法（差分思想）

1. 題目

假設你有一個長度爲 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
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。

2. 解題

類似題目：
LeetCode 1109. 航班預訂統計（差分思想）
LeetCode 1094. 拼車

• 左端點加上數字，右端點+1的位置 減去該數字
• 最後遍歷求前綴和

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

40 ms 14.1 MB

---

我的CSDN博客地址 https://michael.blog.csdn.net/

長按或掃碼關注我的公衆號（Michael阿明），一起加油、一起學習進步！