前缀和 差分 数组

前缀和 差分

前缀和用于求解单点更新后区间的和，差分用于求解区间更新后区间内的单个元素

一维前缀和

将现有数组看作是数组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;
}