1 前缀和数组

适用于快速频繁的计算一个索引区间内的元素之和，核心思想就是使用一个前缀和数组，然后使用前缀和数组的两个元素之差，来快速判断原来的数组的子数组之和，具体如下题所示。

在这里插入图片描述

1.1 题解

前缀和数组的具体思路就是，先new一个新的数组preSum出来，preSum[i]记录的是nums[0…i-1]，也就是nums数组中每一个元素的前缀和。比如preSum记录索引为0的元素的前缀和为2，记录索引为1的元素的前缀和为-2+0=-2，索引为2元素的前缀和为-2+0+3=1以此类推，这样就能达到一个快速计算前缀和的一个技巧。

1.2 Code

class NumArray {
private:vector<int>preSum;
public:NumArray(vector<int>& nums) {preSum.resize(nums.size() + 1);for (int i = 1; i < preSum.size(); ++i){preSum[i] = preSum[i - 1] + nums[i - 1];}}int sumRange(int left, int right) {return preSum[right + 1] - preSum[left];}
};/*** Your NumArray object will be instantiated and called as such:* NumArray* obj = new NumArray(nums);* int param_1 = obj->sumRange(left,right);*/

1.3 结果

在这里插入图片描述

2 二维矩阵的前缀和数组

在这里插入图片描述

2.1 题解

给定一个矩阵的左上角元素以及右下角元素，就能知道矩阵的一个大小了，现在题目给出这两个数值，然后要求给出这个子矩阵的所有元素之和，如上图就是红框内元素之和。因为这是矩阵的加减法，可以用比较大的矩阵减去两个小矩阵再加回来多减的一个部分，就能得到所求的结果。

2.2 Code

class NumMatrix {
// 定义：preSum[i][j] 记录 matrix 中子矩阵 [0, 0, i-1, j-1] 的元素和
private:vector<vector<int>> preSum;
public:NumMatrix(vector<vector<int>>& matrix) {int m = matrix.size(), n = matrix[0].size();if (m == 0 || n == 0) return;//构造前缀和数组preSum.resize(m+1, vector<int>(n+1));for (int i = 1; i <= m; ++i){for (int j = 1; j <= n; ++j){preSum[i][j] = preSum[i-1][j] + preSum[i][j-1] + matrix[i - 1][j - 1] - preSum[i-1][j-1];}}}int sumRegion(int x1, int y1, int x2, int y2) {return preSum[x2+1][y2+1] - preSum[x1][y2+1] - preSum[x2+1][y1] + preSum[x1][y1];}
};/*** Your NumMatrix object will be instantiated and called as such:* NumMatrix* obj = new NumMatrix(matrix);* int param_1 = obj->sumRegion(row1,col1,row2,col2);*/

2.3 结果

在这里插入图片描述

3 差分数组

差分数组主要适用场景主要是对原数组的某个区间的元素进行增减。前缀和是频繁求区间和，差分数组是频繁修改某个区间的元素，同时增加或者减少。比如同时给nums中所有元素加1，再同时减2，最后输出nums。

构造差分数组：

vector<int> diff;
diff.resize(nums.size());
diff[0] = nums[0];//首元素就是nums的首元素
for (int i = 1; i < num.size(); ++i)
{diff[i] = nums[i] - nums[i - 1];//后一项减前一项
}

还原差分数组：

vector<int> res;
res.resize(diff.size());
res[0] = diff[0];
for (int i = 1; i < diff.size(); ++i)
{res[i] = res[i - 1] + diff[i];
}

有了diff数组之后，想要给原数组当中，索引为i到j的数同时加减就很简单了，比如说对nums[i…j]当中进行整体加3，就可以先对diff[i] + 3，再让diff[j+1] -3，这样就可以达到上述效果。

// 差分数组工具类
class Difference {// 差分数组private int[] diff;/* 输入一个初始数组，区间操作将在这个数组上进行 */public Difference(int[] nums) {assert nums.length > 0;diff = new int[nums.length];// 根据初始数组构造差分数组diff[0] = nums[0];for (int i = 1; i < nums.length; i++) {diff[i] = nums[i] - nums[i - 1];}}/* 给闭区间 [i, j] 增加 val（可以是负数）*/public void increment(int i, int j, int val) {diff[i] += val;if (j + 1 < diff.length) {diff[j + 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;}
}