前缀和与差分
前缀和是指某个序列的前 n 项和,可以把它理解为数学上的数列的前 n 项和,而差分可以看成前缀和的逆运算。
一、一维前缀和
一维前缀和主要用于计算某一区间段的和,可以将复杂的问题简单化。一般步骤如下:
先做一个前缀和预处理,定义原数组为 a[]
,前缀和数组为 sum[]
(下标从 1 开始),sum[i]
代表 a
数组中前 i
个数的和,即 sum[i] = a[1] + a[2] + ... + a[i]
.
求前缀和的公式:
sum[i] = sum[i - 1] + a[i]
.
再进行查询操作:
求某段区间 [l, r]
的和: a[l] + ... + a[r] = sum[r] - sum[l - 1]
.
对于每次查询,只需执行 sum[r] - sum[l - 1]
,时间复杂度为 O(1).
【例题】一维前缀和
输入一长度为 n 的整数序列
接下来再输入 m 个询问,每个询问输入一对 l,r
对于每个询问,输出原序列中从第 l 个数到第 r 个数的和
#include <iostream>
using namespace std;
const int N = 100010;
int n, m;
int a[N], s[N];
int main()
{
cin >> n >> m;
for (int i = 1; i <= n; i++){
cin >> a[i];
s[i] = s[i - 1] + a[i]; //计算前缀和
}
while(m--){
int l, r;
cin >> l >> r;
//输出区间段的和
cout << s[r] - s[l - 1] << endl;
}
return 0;
}
二、二维前缀和
常用于求二维矩阵中求某一子矩阵的区域和。一般步骤如下:
与一维前缀和类似,简单来说就是套公式。先求出每一部分的前缀和,再用大区域的前缀和减去小区域的前缀和。
sum[i, j]
表示第 i 行第 j 列 格子左上部分所有元素的和.
以 (x1, y1) 为左上角,(x2, y2) 为右下角的子矩阵的和为:
sum[i][j] - sum[x1 - 1][y2] - sum[x2][y1 - 1] + sum[x1 - 1][y1 - 1]
【例题】二维前缀和
给定一个 n 行 m 列的整数矩阵,q个操作,(x1, y1) 和 (x2, y2) 表示一个子矩阵的左上角和右下角坐标
每次操作要计算出子矩阵的值并输出
#include <iostream>
using namespace std;
const int N = 1010;
int n, m, q;
int a[N][N], s[N][N];
int main()
{
cin >> n >> m >> q;
for (int i = 1; i <= n; i++)
for (int j = 1; j <= m; j++){
cin >> a[i][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;
//输出子矩阵的值
printf("%d\n", s[x2][y2] - s[x1 - 1][y2] - s[x2][y1 - 1] + s[x1 - 1][y1 - 1]);
}
return 0;
}
三、一维差分
差分可以看成前缀和的逆运算。
首先给定一个原数组 a[]
,可以构造一个差分数组 b[]
,使得 a[i] = b[1] + b[2] + ... + b[i]
,那么称 a[]
为 b[]
的前缀和数组,b[]
为 a[]
的差分数组。
差分常用于:比如我们有一个 a[]
,要求把区间 [l, r]
中的每一个数都加上一个常数 c
. 如果暴力做法直接遍历的话,效率较低,使用差分就能高效的解决。
先求其差分,由于 b[]
是 a[]
的差分数组,对 b[i]
的修改,会影响到从 a[i]
往后的每一个数。对于区间 [l, r]
,首先 b[l] + c
,使得 a[l]
及以后(超过 a[r] )的数都加上了 c
,所以还需要 b[r + 1] - c
,使得 a[r]
后面的数保持原样。再求其前缀和,即可在 O(1) 的时间内得到结果。
【例题】一维差分
输入一个长度为 n 的整数序列
接下来输入 m 个操作,每个操作包含三个整数 l,r,c ,表示将序列 [l, r] 之间的每个数都加上 c
请你输出进行完所有操作后的序列
#include <iostream>
using namespace std;
const int N = 100010;
int n, m;
int a[N], b[N];//a是原数组,b是差分数组
int main(){
cin >> n >> m;
for (int i = 1; i <= n; i++){
cin >> a[i];
b[i] = a[i] - a[i - 1]; //构建差分数组
}
while(m--){
int l, r, c;
cin >> l >> r >> c;
b[l] += c; //后面所有部分都会加
b[r + 1] -= c; //再减去多余的部分
}
for (int i = 1; i <= n; i++){
a[i] = b[i] + a[i - 1];
cout << a[i] << " ";
}
return 0;
}
四、二维差分
与一维差分类似,二维差分则用于对二维数组中子矩阵的值进行快速操作。假设 a[][]
为原数组,b[][]
为差分数组:
给以(x1, y1)为左上角,(x2, y2)为右下角的子矩阵中的所有元素加上c:
b[x1][y1] += c;
b[x2 + 1][y1] -= c;
b[x1][y2 + 1] -= c;
b[x2 + 1][y2 + 1] += c;
【例题】二维差分
一个 n 行 m 列的整数矩阵,q 个操作,(x1, y1) 和 (x2, y2) 表示一个子矩阵的左上角和右下角坐标
每个操作要将选中的子矩阵中的每个元素的值加上 c
将进行完所有操作后的矩阵进行输出
#include <iostream>
using namespace std;
const int N = 1010;
int n, m, q;
int a[N][N], b[N][N];//a为原矩阵,b为差分矩阵
//对差分数组进行操作
void insert(int x1, int y1, int x2, int y2, int c){
b[x1][y1] += c;
b[x2 + 1][y1] -= c;
b[x1][y2 + 1] -= c;
b[x2 + 1][y2 + 1] += c;
}
int main(){
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]); //构造差分数组
}
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++)
b[i][j] += b[i - 1][j] + b[i][j - 1] - b[i - 1][j - 1]; //二维前缀和
for (int i = 1; i <= n; i++){
for (int j = 1; j <= m; j++)
printf("%d ", b[i][j]);
puts("");
}
return 0;
}