一、前缀和
1. 前缀和
已知原数组A:a[1] .... a[n]
构造前缀和数组:s[1] .... s[n]
初始化前缀和数组:s[i] = s[i-1] + a[i]
(i从1开始)
使用:计算a[l]~a[r]之间的和
sum = s[r] - s[l-1]
2.二维前缀和
问题一:如何初始化前缀和数组?(见图中红框)
s[i][j] = s[i-1][j] + s[i][j-1] - s[i-1][j-1] + a[i][j]
使用
问题二:如何求以(x1,y1)
为左顶点,(x2,y2)
为右顶点的内部子矩阵数的和?
sum = s[x2][y2] - s[x2][y1-1] - s[x1-1][y2] + s[x1-1][y1-1]
二、差分
1.差分
已知原数组A:a[1] .... a[n]
构造差分数组:b[1] .... b[n]
使得 a[i] = b[1] + b[2] + .... + b[i]
使用:a[l]~a[r] 都 +c
插入操作
b[l] += c;
b[r+1] -= c;
注: 可将a数组看作全为0的数组,对其进行构造即为a[i] = 在 (i,i) 区间内 + a[i]
,这个过程中可以初始化差分数组b。
例题
#include <iostream>
using namespace std;
const int N = 100010;
int a[N],b[N];
//核心操作:插入
void insert(int l, int r, int c)
{
b[l] += c;
b[r+1] -= c;
}
int main()
{
int n,m;
cin >> n >> m;
for(int i = 1; i <= n; i++)
{
cin >> a[i];
insert(i,i,a[i]); //构造出差分数组
}
//完成插入操作
while(m--)
{
int l,r,c;
cin >> l >> r >> c;
insert(l,r,c);
}
for(int i = 1; i <= n; i++)
{
//插入后对已经改变的差分数组求前缀和即可
a[i] = a[i-1] + b[i];
cout << a[i] << " ";
}
return 0;
}
2.二维差分
已知a数组,构造b数组,使得a是b的二维前缀和,即b为a的二维差分数组
使用
以(x1,y1)
为左顶点,(x2,y2)
为右顶点的内部子矩阵的每个数 + c
对于差分数组
b[x1][y1] += c;
b[x1][y2+1] -= c;
b[x2+1][y1] -= c;
b[x2+1][y2+1] += c;
注: 可将a数组看作全为0的数组,对其进行构造即为a[i][j] = (i,j)到(i,j) + a[i][j]
,这个过程中可以初始化差分数组b。
例题
#include <iostream>
using namespace std;
const int N = 1010;
int a[N][N],b[N][N];
void insert(int x1, int y1, int x2, int y2, int c)
{
b[x1][y1] += c;
b[x1][y2+1] -= c;
b[x2+1][y1] -= c;
b[x2+1][y2+1] += c;
}
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];
insert(i,j,i,j,a[i][j]); //构造出b数组
}
while(q--)
{
int x1, y1, x2, y2, c;
cin >> x1 >> y1 >> x2 >> y2 >> c;
insert(x1,y1,x2,y2,c);
}
for(int i = 1; i <= n; i++)
{
for(int j = 1; j <= m; j++)
{
a[i][j] = a[i-1][j] + a[i][j-1] - a[i-1][j-1] + b[i][j]; //求前缀和
cout << a[i][j] << " ";
}
cout << endl;
}
return 0;
}