【acwing算法基础课笔记】01-基础算法-前缀和&差分

一维前缀和

Sr包含 a1 ~ ar 的信息
Sl 包含 a1 ~ al 的信息

求$S_{l到r}$ 只需要利用她两的差，不用逐个遍历

从 1开始 a的数列，S0 可以代表为空的集合而避免边界问题，例如求$[1,10]$ ,就可以直接$[S_{10}-S_0]$，可以用公式$[S_r-S_{l-1}]$代表

#include <iostream>

using namespace std;

const int N = 100010;

int n,m; // n个数，求m次
int a[N],s[N];

/*
 求 a1 ~ an 中区间 [l,r] 的和
*/

int main(){

    // 停止监听标准输入，优点：提高cin输入速度，缺点：无法使用scanf
    ios::sync_with_stdio(false);
    
    cin >> n >> m;
    for(int i=1;i<=n;i++)
        cin >> a[i];
    for(int i=1;i<=n;i++)
        s[i] = s[i-1];
    
    while(m--){
        int l,r;
        cin >> l >> r;
        cout << s[r] - s[l-1];
    };

    return 0;
}

二维前缀和

可以扩展到二维

求前缀和序列

求区间和

#include <iostream>
#include <vector>

using namespace std;

const int N =1010;

int n,m,q; // n m 二维矩阵大小 q 求几次区间和
int a[N][N],s[N][N];
/*
    二维前缀和差分
*/
int main(){
    cin >> n >> m >> q;
    for(int i=1;i<=n;++i)
        for(int j=1;j<m;++j)
            cin >> a[i][j];
    
    // 求前缀和
    for(int i=1;i<=n;++i)
        for(int j=1;j<m;++j)
            s[i][j] = s[i-1][j]+s[i][j-1] - s[i-1][j-1] + a[i][j];

    // 求部分和
    while(q--){
        int x1,y1,x2,y2;
        cin >> x1 >> y1 >> x2 >> y2;
        cout << s[x2][y2] - s[x1][y2]- s[x2][y1] + s[x1][y1]<< endl;
    }

}

一维差分

S是原数组（假设他是b的前缀和数组），则对S的某段区间的操作O(N)可以转为对其映射的b前缀和数组操作O(1),可以多次操作O(1)后转为原数组

得到公式
S[l~r] + C = b[l] + C and b[r+1] - C

作用 用O(1)的时间给原数组（差分数组）加上值

差分数组的构建

利用上述公式构建
将前缀和数组和差分看作初始为0的数组，对原数组[i，i]这段进行+C操作，C为$a_i$ 。也就是对差分数组 b[i] + C and b[i+1]-C

#include <iostream>

using namespace std;

const int N = 100010;

int n,m;
int a[N],b[N];

void myInsert(int l,int r,int c){
    b[i] += c;
    b[r+1] -= c;
}
/*
 利用前缀和与差分数组的性质来对某一区间直接+C
*/
int main(){
    cin >> n >> m;
    // 前缀和数组
    for(int i=1;i<=n;++i)
        cin>> a[i];
    // 差分数组构建，从前缀和数组得出差分数组
    for(int i=1;i<n;++i)
        myInsert(i,i,a[i]); 
    // 从差分数组 求 前缀和数组
    for(int i=1;i<=n;++i) b[i]+=b[i-1];
    // 输出
    for(int i=1;i<=n;++i) 
        cout << b[i] << endl;
}

二维差分

【2:06】

要使 a【x1,y1】 ~ a【x2,y2】间的面积元素操作，可以直接对其映射的差分数组b操作

图为b的矩阵，阴影为对单个b+c所影响的a数组范围

从图易得
ax1y1~ax2y2 +C 等效于
$b_{x_1,y_1}+=C$
$b_{x_2+1,y_1}-=C$
$b_{x_1,y_2+1}-=C$
$b_{x_2+1,y_2+1}+=C$

#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[x1+1][y1] -= c;
    b[x1][y1+1] -= c;
    b[x1+1][y1+1] += c;
}

int main(){
    // 输入 a 前缀和数组
    cin >> n >> m >> q;
    for(int i=1;i<=n;++i)
        for(int j=1;j<=m;++j)
            cin >> a[i][j];
    
    // 构建 b 差分数组
    for(int i=1;i<=n;++i)
        for(int j=1;j<=m;++j)
            insert(i,j,i,j,a[i][j]);
    
    // 对a的区间整体操作O(n)映射到b的几个点操作,n次O(1)操作
    while(q--){
        int x1,y1,x2,y2,c;
        cin >> x1 >> x2 >> x2 >> y2 >> c;
        insert(x1,y1,x2,y2,c);
    }
    // 转换 b 数组到 a 输出
    for(int i=1;i<=n;++i)
        for(int j=1;j<=m;++j){
            b[i][j] += b[i-1][j-1] - b[i-1][j] - b[i][j-1] + b[i-1][j-1];
            cout << b[i][j];
        }

}