前缀和 差分
前缀和用于求解单点更新后区间的和,差分用于求解区间更新后区间内的单个元素
一维前缀和
将现有数组看作是数组a ,求解A的前缀和数组S
S[i] = a[1] + a[2] + ... a[i]
a[l] + ... + a[r] = S[r] - S[l - 1]
二位前缀和
S[i, j] = 第i行j列格子左上部分所有元素的和
以(x1, y1)为左上角,(x2, y2)为右下角的子矩阵的和为:
S[x2, y2] - S[x1 - 1, y2] - S[x2, y1 - 1] + S[x1 - 1, y1 - 1]
#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=1010;
int n,m,q,s[maxn][maxn],x,y,x2,y2;
int main(void){
scanf("%d%d%d",&n,&m,&q);
for(int i=1;i<=n;++i){
for(int j=1;j<=m;++j){
scanf("%d",&s[i][j]);
}
}
//计算前缀和数组
for(int i=1;i<=n;++i){
for(int j=1;j<=m;++j){
s[i][j]=s[i][j]+s[i][j-1]+s[i-1][j]-s[i-1][j-1];
}
}
//前缀和矩阵相减
while(q--){
scanf("%d%d%d%d",&x,&y,&x2,&y2);
printf("%d\n",s[x2][y2]-s[x-1][y2]-s[x2][y-1]+s[x-1][y-1]);
}
}
一维差分
将现有数组看作是数组a的前缀和数组S,利用差分求解a
a是已知数组S的对应差分数组,也就是说a做前缀和得到S
给区间[l, r]中的每个数加上c:a[l] += c, a[r + 1] -= c
差分数组的初始化
初始化看作是对差分数组的更新,在全0的数组中的[i,i]区间加上s[i]的值
for(int i=1;i<=n;++i) insert(i,i,s[i]);
区间更新 【l,r】上加上c
void insert(int l,int r,int c){
a[l]+=c;
a[r+1]-=c;
}
二维差分
a是差分数组
给以(x1, y1)为左上角,(x2, y2)为右下角的子矩阵中的所有元素加上c:
a[x1, y1] += c, a[x2 + 1, y1] -= c, a[x1, y2 + 1] -= c, a[x2 + 1, y2 + 1] += c
void insert(int x1,int y1,int x2,int y2,int c){
a[x1][y1]+=c;
a[x1][y2+1]-=c;
a[x2+1][y1]-=c;
a[x2+1][y2+1]+=c;
}