一维前缀和
有一串长度为n的数列an,如果要求我们访问**[j,k]**里的数的和。
我们一般的方法是分别求出Sj和Sk的值然后让他们相减。
如果我们了解到前缀和,我们就可以使用下面这种方法
for(int i=1;i<=n;i++) a[i]+=a[i-1]; //a[i]相当于Si(前i项和)
前缀和就是前面i个数的总和
接下来,我们只需要分别找到a[j-1] 和 a[k],sum=a[k]-a[j-1] (包括a[k]和a[j])
二维前缀和
规定:i为行数,j为列数
如果我们想要去求得a[i][j]的前缀和,
S[i][j]=S[i-1][j]+S[i][j-1]+S[i-1][j-1];
求a[2][4]的前缀和 代码如下:
#include<iostream>
using namespace std;
const int m=10;
int a[m][m]; //定义全局变量多余的空自动填充为0
void func()
{
int i,j;
for(i=1;i<=5;i++){
for(j=1;j<=5;j++) cin>>a[i][j];
}
for(i=1;i<3;i++){
for(j=1;j<5;j++){
a[i][j]+=a[i-1][j]+a[i][j-1]-a[i-1][j-1];
}
}
cout<<a[2][4];
}
int main()
{
func();
return 0;
}
通过二维前缀和求某一矩阵和
从a[2][2] (左上角) 到 a[3][4] (右下角)
#include<iostream>
using namespace std;
const int m=10;
int a[m][m];
void func()
{
int i,j,ans;
for(i=1;i<=5;i++){
for(j=1;j<=5;j++) cin>>a[i][j];
}
for(i=1;i<6;i++){
for(j=1;j<6;j++){
a[i][j]+=a[i-1][j]+a[i][j-1]-a[i-1][j-1];
}
} //该循环实现了每一点都变成了二维前缀和
//已知(x1=2,y1=2) (x2=3,y2=4) 求这两个点构成的矩阵和
//S( 2,2 -> 3,4)=S[3,4]-S[x1-1,y2]-S[x2,y1-1]+S[x1-1,y1-1]
ans=a[3][4]-a[1][4]-a[3][1]+a[1][1];
cout<<ans;
}
int main()
{
func();
return 0;
}
1 2 3 4 5
6 7 8 9 10
11 12 13 14 15
16 17 18 19 20
21 22 23 24 25