目录
一维前缀和
S0=0;
Si=a1+a2+……+ai;
i从1开始相比于从0开始可以减少特判,且sum[1,r]=Sr
作用:快速求数组中[l,r]段和sum=s[r]-s[l-1];
输入一个长度为 n的整数序列。
接下来再输入 m个询问,每个询问输入一对 l,r。
对于每个询问,输出原序列中从第 l个数到第 r个数的和。
#include<algorithm>
#include<iostream>
using namespace std;
const int N = 1e6;
int a[N];
int s[N] = { 0 };
int main()
{
int n, q;
cin >> n >> q;
for (int i = 1; i <= n; i++)
{
cin >> a[i];
s[i] = s[i - 1] + a[i];
}
while (q--)
{
int t1, t2;
cin >> t1 >> t2;
cout << s[t2] - s[t1 - 1] << endl;
}
return 0;
}
二维前缀和
Sij表示其左上角全部元素和
Sij=S(i-1,j)+S(i,j-1)-S(i-1,j-1)+aij;
作用:快速求子矩阵和
S阴影=S(x2 y2)-S(x2 y1-1)-S(x1-1 y2)+S(x1-1 y1-1) (注:这里面积实际是离散的,化成表格会更好理解)
输入一个 n 行 m 列的整数矩阵,再输入 q个询问,每个询问包含四个整数 x1,y1,x2,y2,表示一个子矩阵的左上角坐标和右下角坐标。
对于每个询问输出子矩阵中所有数的和。
#include<algorithm>
#include<iostream>
using namespace std;
const int N = 1010;
int s[N][N];
int a[N][N];
int main()
{
int n, m, q;
cin >> n >> m >> q;
for (int i = 1; i <= n; i++)
{
for (int j = 1; j <= m; j++)
{
cin >> a[i][j];
}
}
for (int i = 1; i <= n; i++)
{
for (int j = 1; j <= m; j++)
{
s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1] + a[i][j];
}
}
while (q--)
{
int x1, y1, x2, y2;
cin >> x1 >> y1 >> x2 >> y2;
cout << s[x2][y2] - s[x2][y1 - 1] - s[x1 - 1][y2] + s[x1-1][y1-1]<<endl;
}
return 0;
}