差分
定义:差分与前缀和相反,给定一系列数a1,a2,a3…an,构造b1,b2,b3…bn使得对于任意的ai,
有ai=b1+b2+b3+…+bi,即为a数组是b数组的前缀和,则b数组是a数组的差分,差分的构造方式:
b1=a1;
b2=a2-a1;
b3=a3-a3;
…
bi=bi-bi-1;
作用:可以用O(1)的时间复杂度为原数组a在某个区间[l.r]内全部添加一个值,(直接的添加的话时间复杂度是O(n),而使用a的差分b只需修改两处,bl+c,br-c,由于bl+c,故al和al后边的所有元素都会加上c,同理,br+1 -c,所以ar+1和ar+1后边的所有元素都会减去c,这样的话,就只有区间[L,R]上的a的元素加上了C,后边的会被+c,-c抵消。)
题目: 输入一个长度为 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
输入样例:
6 3
1 2 2 1 2 1
1 3 1
3 5 1
1 6 1
输出样例:
3 4 5 3 4 2
代码如下:
#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;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
//构造差分序列b
for(int i=1;i<=n;i++)
{
insert(i,i,a[i]);
}
//在一定区间上进行+c的处理,即为对差分序列b处理
while(m--)
{
int l,r,c;
scanf("%d%d%d",&l,&r,&c);
insert(l,r,c);
}
//求前缀和,用差分序列的前缀和表示处理后的a[i],
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;
}
main函数中的第二个循环是为了构造差分序列b,采取的方式是将原数组a视为所有元素全为0的数组,然后通过在区间[i,i]上插入元素a[i](insert(i,i,a[i])的方式进行构造,得到差分序列b:
b1=a1;
b2=a2-a1;
b3=a3-a3;
…
bn=bn-bn-1;
差分矩阵
如图,矩阵的每一个格子表示二维数组b的对应位置的元素,比如在b[x1][y1][的的位置上+c,表示从这个角开始的右下的方块内的所有的(x,y)对应的axy都会加上c,那么,要仅仅让(x1,y1)到(x2,y2)的区域上axy加c,就必须把其余的部分减去c,因为要减去的两个部分有重合部分,重合部分的c被减了两次,所以要在给重合区域加上一个c,处理这个操作的公式如下:
b[x1][y1]+=c;
b[x1][y2+1]-=c;
b[x2+1][y1]-=c;
b[x2+1][y2+1]+=c;
差分矩阵首先要理解b[i][j]+c是(i,j)右下的区域上的所有点所对应的子矩阵的和都会+c;
然后分别计算不同区域所代表的值:如图红色区域为b[x1,y2+1]表示要减去c的区域;
如图红色区域2表示另一个需要减去c的区域:b[x2+1][y1];
由于b[x2+1][y2+1]被减去了两次c,所以要再加回来一次c;
所以最终的公式是:
b[x1][y1]+=c;
b[x1][y2+1]=c;
b[x2+1][y1]-=c;
b[x2+1][y1+1]+=c;
二维的建议画图理解一下,更好理解;
构造b[i][j]的思想与一维差分的思想相同,题目代码如下:
需要注意的是构造差分矩阵的时候,直接使用写好的插入函数insert(i,j,i,j,a[i][j])即可;
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
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[x1][y2+1]-=c;
b[x2+1][y1]-=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]);
int x1,y1,x2,y2,c;
while(q--){
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][j-1]+b[i-1][j]-b[i-1][j-1];
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++) cout<<b[i][j]<<" ";
cout<<endl;
}
return 0;
}