基础-前缀和 差分

前缀和差分是一对逆运算

前缀和公式就是Si=

三个问题：

1. 如何求Si:注意S0=0
2. 什么作用:快速求出一段数的和

一维前缀和

公式

S[i] = a[1] + a[2] + ... a[i]
a[l] + ... + a[r] = S[r] - S[l - 1]

代码

#include <iostream>
#include <cstring>
#include <algorithm>
const int N = 100000+10;
int n,m,q[N];
int main()
{
    scanf("%d%d", &n, &m);
    q[0]=0;
    for (int i = 1; i <= n; i ++ ){
        int x;
        scanf("%d", &x);
        q[i]=q[i-1]+x;
    }
    while (m -- ){
    int l,r;
    scanf("%d%d", &l,&r);
    int ans=q[r]-q[l-1];
    printf("%d\n",ans);
    }
}

二维前缀和

公式

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]

代码

注意空出所有索引为0的地方，方便统一处理。

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1010,M = 1010;
int n,m,q,s[N][M]={0};
int main()
{
    scanf("%d%d%d", &n, &m,&q);
    for (int i = 1; i <= n; i ++ )
        for (int j = 1; j <= m; j ++ ){
            int x;
            scanf("%d", &x);
            s[i][j]=s[i][j-1]+s[i-1][j]-s[i-1][j-1]+x;
            
        }
    while (q--){
        int x1,y1,x2,y2;
        cin>>x1>>y1>>x2>>y2;
        cout<<s[x2][y2]-s[x2][y1-1]-s[x1-1][y2]+s[x1-1][y1-1]<<endl;
    }
}

差分

概念：构造一个数组b，使原数组a是他的前缀和

用处：在o(n)的时间内由b得a

        让a【l：r】元素+c，直接处理可能导致n的复杂度；

        在b数组只需要修改俩数即b【l】+=c,b【r+1】-=c；

构造：假定a数组初始全是0，给定的a数组值可以看作n次插入操作，第i次操作是insert(i,i,a[i])

代码

#include <iostream>
#include <cstring>
#include <algorithm>
const int N = 100000+10,M=100000+10;
int n,m;
int a[N],b[N];
void inser(int l,int r,int 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 ++ ) inser(i,i,a[i]);//初始化b差分数组
    
    while (m -- ){
        int l,r,c;
        scanf("%d%d%d", &l,&r,&c);
        inser(l,r,c);
    }
    for(int i=1;i<=n;i++) b[i]+=b[i-1];
    for (int i=1;i<=n;i++) printf("%d ",b[i]);
}

#include<iostream>
using namespace std;
const int N=1e6+10;
int n,m;
int a[N];
//这是一种简化的复合写法，只需要维护一个差分数组A即可
void h(int l,int r,int c){
    a[l]+=c;
    a[r+1]-=c;
    
}
int main(){
    cin>>n>>m;
    for (int i=1;i<=n;i++)
        {//在读入数据的时候，就把他初始化为差分数组
            int x;
            scanf("%d",&x);
            h(i,i,x);
        }
    while (m -- ){//插入操作
        int l,r,c;
        cin>>l>>r>>c;
        h(l,r,c);
    }
    //将差分数组求前缀和，而后打印，相当于是原数组处理之后的结果
    for (int i = 1; i <= n; i ++ ) {a[i]+=a[i-1];printf("%d ",a[i]);}
    
    
}

 

二维差分

跟一维的一样，初始a为0，输入的aij视为在i*j的区域内插入a[i][j]，无论是b的构造还是后面的修改操作，都只需要进行插入操作。

公式：给以(x1, y1)为左上角，(x2, y2)为右下角的子矩阵中的所有元素加上c：
S[x1, y1] += c, S[x2 + 1, y1] -= c, S[x1, y2 + 1] -= c, S[x2 + 1, y2 + 1] += c

 代码

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1010,M=1010;
int m,n,q;
int b[N][M]={0};
void up(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()
{
    cin>>n>>m>>q;
    for (int i = 1; i <= n; i ++ )
        for (int j = 1; j <= m; j ++ )
            {//边读数据边创建差分矩阵b
                int x;
                scanf("%d", &x);
                up(i,j,i,j,x);
            }
    while (q -- ){//数次更新操作
        int x1,y1,x2,y2,c;
        cin>>x1>>y1>>x2>>y2>>c;
        up(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]);
            cout <<b[i][j]<<" ";
        }
        cout<<endl;
    }
}