建议大家可以跟着前缀和一块学习。可以参考我上一篇“前缀和”博客,希望可以帮助到大家。
一、 一维基本知识
概念
- 前缀和的逆运算
- 原数组:
a1, a2, a3,...an
构造b数组b1, b2, ...bn
使得:
ai = b1 + b2 + b3 + ... + bi
- 了解一维的构造
b1 = a1;
b2 = a2 - a1;
b3 = a3 - a2;
...
bn = an - an-1
- a称为b的前缀和, b称为a的差分
用处
- 对差分数组求一次前缀和便可求出原数组,只要有b数组,便可在
O(n)时间,求出a数组 - 给定区间
[l,r]
使得原数组a的该区间内全部加上C
操作原理
目标: a数组的[l, r] 区间内加C
(1)让 b[l] = b[l] + C
根据定义,a[l], a[l + 1], a[l + 2], a[l + 3],...a[n]
均会加C。
(2)让b[r + 1] = b[r + 1] - C
这样, a[r+1] .. a[n]
会把(1)中多加的C减掉。
用O(1)的时间,便会给原数组某段区间加上一个值
对于b数组,可以看成a数组的n次插入操作, 初始a,b均为0,a[1] 意味着在[1,1]区间内加上a1, ...an意味着在[n,n]区间内加上n
二、一维的一道模板题
题目描述:
输入一个长度为n的整数序列。
接下来输入m个操作,每个操作包含三个整数l, r, c,表示将序列中[l, r]之间的每个数加上c。
请你输出进行完所有操作后的序列。
输入格式
第一行包含两个整数n和m。
第二行包含n个整数,表示整数序列。
接下来m行,每行包含三个整数l,r,c,表示一个操作。
输出格式
共一行,包含n个整数,表示最终序列。
数据范围
1≤n,m≤100000,
1≤l≤r≤n,
−1000≤c≤1000,
−1000≤整数序列中元素的值≤1000
#include <iostream>
using namespace std;
const int N = 100010;
int n, m;
int a[N], b[N];
void insert(int l, int r, int c){ //在l, r之间加上C
b[l] += c;
b[r + 1] -= c;
}
int main(){
scanf("%d %d", &n, &m);
for(int i = 1; i <= n; i ++) scanf("%d", &a[i]);
for(int i = 1; i <= n; i ++) insert(i, i, a[i]);
while(m --){
int l, r, c;
scanf("%d%d%d", &l, &r, &c);
insert(l, r, c);
}
for(int i = 1; i <= n; i ++) b[i] += b[i - 1];
//对差分数组求一遍前缀和,便是改变后的a数组
for(int i = 1; i <= n; i ++) printf("%d ", b[i]);
return 0;
}
三、二维差分
概念
原来a[i][j]
是差分矩阵b[i][j]
的前缀和
操作原理
如何给一个子矩阵加一个值?
目标:
在a矩阵中, (x1,y1)为左上角、(x2, y2)为右上角 的区域内加上C
(1) b[x1, y1] += C,
(2)b[x2 + 1, y1] -= C
绿色区域
(3) b[x1, y2+1] -= C
红色区域
(4)b[x2+1, y2+1] += C
减重复的加上C
初始化:插入a[i, j]操作可以看成,插入一个(i,j),(i,j)的小矩阵
四、二维的一道模板题
题目描述:
输入一个n行m列的整数矩阵,再输入q个操作,每个操作包含五个整数x1, y1, x2, y2, c,其中(x1, y1)和(x2, y2)表示一个子矩阵的左上角坐标和右下角坐标。
每个操作都要将选中的子矩阵中的每个元素的值加上c。
请你将进行完所有操作后的矩阵输出。
输入格式
第一行包含整数n,m,q。
接下来n行,每行包含m个整数,表示整数矩阵。
接下来q行,每行包含5个整数x1, y1, x2, y2, c,表示一个操作。
输出格式
共 n 行,每行 m 个整数,表示所有操作进行完毕后的最终矩阵。
数据范围
1≤n,m≤1000,
1≤q≤100000,
1≤x1≤x2≤n,
1≤y1≤y2≤m,
−1000≤c≤1000,
−1000≤矩阵内元素的值≤1000
#include <iostream>
using namespace std;
const int N = 1010;
int n, m, q;
int a[N][N], b[N][N];
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(){
scanf("%d %d %d", &n, &m, &q);
for(int i = 1; i <= n; i ++){
for(int j = 1; j <= m; j ++){
scanf("%d", &a[i][j]);
}
}
for(int i = 1; i <= n; i ++){
for(int j = 1; j <= m; j ++){
insert(i, j, i, j, a[i][j]);
}
}
while(q --){
int x1, y1, x2, y2, c;
scanf("%d%d%d%d%d", &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;
}