前缀和与差分

最新推荐文章于 2022-11-04 23:11:18 发布

七个披萨

最新推荐文章于 2022-11-04 23:11:18 发布

阅读量246

点赞数

分类专栏：数据结构和算法文章标签：算法

本文链接：https://blog.csdn.net/weixin_38384296/article/details/125175812

版权

数据结构和算法专栏收录该内容

12 篇文章 0 订阅

订阅专栏

本文介绍了前缀和的概念，包括一维和二维前缀和，并提供了LeetCode练习题作为实例。一维前缀和是数组前n项的和，二维前缀和涉及矩阵的累加和。差分是前缀和的逆运算，常用于区间加法操作。文章通过具体例子展示了差分在解决LeetCode问题中的应用。

摘要由CSDN通过智能技术生成

B站视频：https://www.bilibili.com/video/BV1r4411k7dL?spm_id_from=333.337.search-card.all.click

前缀和

前缀和是一种重要的预处理，能大大降低查询的时间复杂度。可以简单理解为“数列的前n项的和”。根据数组的维度，前缀和有一维前缀和和二维/多维前缀和。

一维前缀和

一维前缀和就是一维数组前n项的和。求解一维前缀和递推公式如下：
$\begin{cases} a[0], & \text{i=0} \\ sum[i-1]+a[i], & \text{i>0} \end{cases}$
当我们求出前缀和数组后，我们可以使用O(1)的时间复杂度求出区间[q1, q2]的区间和：

interval[q1, q2] = sum[q2] - sum[q1 - 1]

求解前缀和数组的java代码如下：

int[] sum = new int[nums.length];
sum[0] = nums[0];
for (int i = 1; i < nums.length; i++) {
    sum[i] = sum[i - 1] + nums[i];
}

有的地方会设置前缀和数组第0项sum[0]=0，递归公式为sum[i]=sum[i-1]+a[i-1]：
int[] sum2 = new int[nums.length + 1];
for (int i = 0; i < nums.length; i++) {
    sum2[i + 1] = sum[i] + nums[i];
}

LeetCode练习题

560. 和为 K 的子数组

523. 连续的子数组和

974. 和可被 K 整除的子数组

二维前缀和

二维前缀和就是二维数组的前缀和，sum[i][j]为上顶点(0, 0)到下顶点(i, j)组成矩阵的各项和：
在这里插入图片描述

二维前缀和的递推公式如下：
$\begin{cases} a[0][0], & \text{i=0, j=0} \\ sum[0][j-1]+a[0][j], & \text{i=0, j>0} \\ sum[i-1][0]+a[i][0], & \text{i>0, j=0} \\ sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+a[i][j], & \text{i>0, j>0} \end{cases}$
图中红色区间sum[i-1][j-1]为重复加的部分，因此需要减去。

当我们求出二维数组的前缀和后，同样可以使用O(1)的时间复杂度求出区间[(q1, q2), (q3, q4)]的各项和：
在这里插入图片描述

interval[(q1, q2), (q3, q4)] = sum[q3][q4] - sum[q1 - 1][q4] - sum[q3][q2 - 1] + sum[q1 - 1][q2 - 1]

LeetCode练习

304. 二维区域和检索 - 矩阵不可变

差分

差分和前缀和可以说是一对“好基友”，差分是前缀和的逆运算，如果存在原数组a[i]，那么对差分数组求前缀和就是原数组：
$\begin{cases} a[0], & \text{i=0} \\ a[i]-a[i-1], & \text{i>0} \end{cases}$
一维差分一般用于这样的场景：多次对序列的某个区间加上一个数，并在最后询问序列是什么样的。如LeetCode1094题拼车。这类题目一般会先对序列求出差分数组，如果题目要求原数组区间[q1, q2)+ 2，那么差分数组只需要操作下面两步：