前缀和
前缀和用于解决数组中的[不同区间的和]的大量查询
核心思路
核心思路是把多个不同区间的问题,统一为相同类型的子问题
也就是把[左右界不固定的问题] 转化为两个 [左边界固定,右边界不定]的问题
- 具体而言,
[l,r]
区间的查询,左右两边界是不固定的,我们固定一边 - 定义
preSum[i]
保存num[0...i)
的所有元素和 - 问题转化为:
[l,r]
区间和=[0,r+1)
区间和 -[0,l)
区间和
(所有[0,i)
区间的和提前求出)
基本应用
模板题
区间和的查询:给出一个数组,多次查询不同区间[l,r]
的区间和
- 原来每个区间的查询问题都是独立的,需要分别计算结果;
而使用前缀和之后,任意区间的查询可以直接通过前缀和得出
LeetCode 1371. 每个元音包含偶数次的最长子字符串
利用前缀和获得任意子串中的元音字符数量,以及进一步优化
类似前缀和,【区间的异或运算】也能通过【前缀异或】来得到,因为a^a=0
LeetCode 1310. 子数组异或查询
对不同区间内数字的异或运算结果查询
解:由于a^(a^b) = 0^b=b
,类似前缀和的处理即可
二维矩阵情况的前缀和
- 需要大量求二维数组内不同矩阵的元素和时,也可以推广到二维矩阵的前缀和:
preSum[i][j]
代表从(0,0)
到(i,j)
区域的元素和(不含i,j)
LeetCode 304. 二维区域和检索 - 矩阵不可变
给出矩阵,多次查询不同子矩阵区域内的元素和(预处理二维前缀和即可)
LeetCode 1314. 矩阵区域和
对于矩阵中每个元素周围k层内的所有元素求和,i,j超出边界时,取边界
LeetCode 661. 图片平滑器
将二维数组中,每各元素与其周围9个点求和,取平均,返回新的数组(注意,在边界的元素,其周围没有9个元素,取平均时要注意)
与之前同样的,有二维的前缀异或问题
LeetCode 1738. 找出第 K 大的异或坐标值
矩阵中,从(0,0)
到(i,j)
的所有元素做异或,能够得到一个值,对于所有可能的i
和j
做这个运算,现在求所有这些结果中,第k大的值
变式
LeetCode 1749. 任意子数组和的绝对值的最大值
要求找到一个子数组,这个子数组和的绝对值最大,问题转化为abs(preSum[j]-preSum[i])
最大,其中i<j
-> 把preSum
看作一个图线,就是要求这条图线的最大高度差