前缀和
一维前缀和
现在有这样一个数组
现在要求求出[1,5]、[2,8]、[3,7]区间元素的和
很简单,就几个for循环就搞定了嘛
1+2+3+4+5
2+3+4+5+6+7+8
3+4+5+6+7
的确,但是当数据规模比较庞大的时候,一个个for得算到猴年马月
观察一下上面几个式子发现,他们之中都有重复的计算,例如3+4+5这一段
那如何能减少重复计算呢?这里提出一个“前缀和”的算法,故名思意,就是每个数都对应保存他前面数的和。
来算一下上面的例子的前缀和
如果要求样例数组在[0,8]区间元素的和就只需要输出sum[7]即可
那如果我要求中间某一段区间内的和呢?例如[3,7]
[3,7]的和就是:3+4+5+6+7与sum[6]差在1+2,欸,这不就是sum[1]吗?那我用sum[6]-sum[1]就可以得到想要的值了
小结:
[0,i]区间和:sum[i]
[i,j]区间和:sum[i]-sum[j-1] i>0;sum[i] i=0
代码:
在这里插入代码片
二维前缀和
第一种想法:转化为一维前缀和
二维数组可以看成n行一维数组拼在一起
例如:求从(1,1)到(2,3)区域的和
那就可以看成两行的区间和相加:(6+7+3)+(3+2+4)将二维前缀和转化为一维前缀和
第二种想法:常规做法
首先,这里定义a[i][j]的前缀和为“以(i,j)为右下角区域的和
得到前缀和之后要怎么求呢?
现在要求黄色部分的区间和,就是整块区域和减去两个绿色的区域和
整块区域和:sum[2][3]
浅绿色区域和:sum[0][3]
深绿色区域和:sum[2][0]
但是要注意的是,在减去两个绿色区域的和的时候,两个绿色重合的位置([0,0])被减了两次,因此要加回一次
也就就是:sum[2][3]-sum[0][3]-sum[2][0]+sum[0][0]
代码:
//计算前缀和
void pre_sum(){
sum[0][0]=g[0][0];
//处理第一列
for(int i=0;i<n;i++)
sum[i][0]=g[i][0]+sum[i-1][0];
//处理第一行
for(int i=0;i<m;i++)
sum[0][i]=g[0][i]+sum[0][i-1];
//处理其他位置的数
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
sum[i][j]=g[i][j]+sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1];
}
}
//获取区间和
int get_sum(int x1,int y1,int x2,int y2){
//左上角的点为(0,0)
if(!x1&&!y1)
return sum[x2][y2];
//左上角的点在第一行
if(!x1)
return sum[x2][y2]-sum[x2][y1-1];
//左上角的点在第一列
if(!y1)
return sum[x2][y2]-sum[x1-1][y2];
return sum[x2][y2]-sum[x2][y1-1]-sum[x1-1][y2]+sum[x1-1][y1-1];
}