前缀和(一维和二维)

前言

前缀和是指某序列的前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];

  • 12
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值