前言
前缀和是指某序列的前n项和,可以把它理解为数学上的数列的前n项和,而差分可以看成前缀和的逆运算。将差分数组进行前缀和就可以得到原数组,合理的使用前缀和与差分,可以将某些复杂的问题简单化,降低程序的时间复杂度。
1.一维前缀和
一维前缀和就是一维数组的数组的前n项和,具体就是用一个数组sum[],来存储数组a[]的前n项和,sum[i]就代表数组a的前 i 项和
#define N 1e5+10
int sum[N], a[N]; //sum[i]=a[1]+a[2]+......+a[i];
for(int i = 1;i <= n; i++){
sum[i]=sum[i-1]+a[i];
}
如果我们要查询区间[L,R]的数组a的和,正常情况下我们要遍历数组a,时间复杂度为O(n),但我们可以直接用sum数组,时间复杂度为O(1)。(不知道什么是时间复杂度的:点这里)
scanf("%d%d",&L,&R);
printf("%d\n", sum[R] - sum[L - 1]);
解释:sum[R]为前R个元素的和,sum[L-1]为前L-1个元素的和,sum[R]-sum[L-1]就为区间[R,L]的和。
原理解释:
sum[r] = a[1] + a[2] + a[3] + a[l-1] + a[l] + a[l + 1] ...... a[r];
sum[l - 1] = a[1] + a[2] + a[3] + a[l - 1];
sum[r] - sum[l - 1] = a[l] + a[l + 1] + ......+ a[r];
2.二维前缀和
如果数组变成了二维的,我们该如何求它的前缀和呢
同一维前缀和一样,我们先来定义一个二维数组s[][],s[i][j] 表示二维数组中,左上角(1,1)到右下角(i, j)
所包围的矩阵元素的和。
先看一张图:
紫色面积是指(1, 1)
左上角到(i, j - 1)
右下角的矩形面积, 绿色面积是指(1, 1)
左上角到(i - 1, j )
右下角的矩形面积。每一个颜色的矩形面积都代表了它所包围元素的和。
所以整个蓝色矩形的面积s[i][j] = 绿色矩形面积s[i-1][j] + 紫色面积s[i][j-1] - 多余红色面积s[i-1][j-1] + 小方块的面积a[i][j];
所以可以得出前缀和公式:s[i][j] = s[i - 1][j] + s[i][j - 1 ] + a[i] [j] - s[i - 1][j - 1]
那么我们现在去求以(x1,y1)
为左上角和以(x2,y2)
为右下角的矩阵的元素的和。
如图:
紫色面积是指 (1, 1)
左上角到(x1 - 1, y2)
右下角的矩形面积 ,黄色面积是指(1, 1)
左上角到(x2, y1 - 1)
右下角的矩形面积;
绿色矩形的面积 = 整个外围面积s[x2, y2] - 黄色面积s[x2, y1 - 1] - 紫色面积s[x1 - 1, y2] + 重复减去的红色面积 s[x1 - 1, y1 - 1]
所以二维前缀和的公式为:
以(x1, y1)为左上角,(x2, y2)为右下角的子矩阵的和为:
s[x2, y2] - s[x1 - 1, y2] - s[x2, y1 - 1] + s[x1 - 1, y1 - 1];