【基础算法】前缀和和差分

模板题

0x01：前缀和模板

题目大意：给你一个数组，在给你两个下标（l, r），需要你返回 l 到 r 的所有数据之和

前缀和：
给定数组g[5] = {0, 1, 2, 3, 4, 5} // 一般从下表1开始
那他的前缀和数组为prefix[5] = {0， 1， 3， 6， 10， 15}

前缀和性质：
0x01：O(1)获取某个范围的数据之和 sum(1, 5) = prefix[5] - prefix[0] // 注意要往前移一格

#include <iostream>

using namespace std;

int n,m;
const int N = 100010; 
int g[N];

int main()
{
    cin >> n >> m;

    for(int i =1;i <= n;i++){
        cin >> g[i];
        g[i] = g[i - 1] + g[i]; // 获取前缀和数组，从1开始（减少边界处理），从前往后
    }

    while(m--){
        cin >>s >> e;
        cout << g[i] - g[i - 1]  << endl; //输出数据
    }
    
    return 0;
}

0x02：差分模板

题目大意：给你一个数组，给你多组数据（l, r, c），需要你把（l, r）中的所有数据都加c

差分：
给定数组g[5] = {0, 1, 2, 3, 4, 5} // 一般从下表1开始
那他的前缀和数组为prefix[5] = {0， 1， 1， 1， 1， 1}

差分性质：
①：获取差分数组，从后面循环获取
②：需要将原数组中（l, r）内的数据都加上c，那就对diffence[l] += c;diffence[r + 1] -= c; // 注意：因为最后一个会超出，但是对数组没有影响，但也要多开辟几个空间，防止出翔边界问题

#include<iostream>

using namespace std;

const int N = 100010;

int g[N];

int n, m;

int main(){
    cin >> n>> m;
    for(int i = 1; i <= n; ++i){
        cin >> g[i];
    }
    
    for(int i = n ; i > 0 ; --i){ // 差分数组，从后往前处理
        g[i] = g[i] - g[i - 1];
    }
    
    while(m--){
        int l, r, c;
        cin >>l>>r>>c;
        g[l] += c;
        g[r + 1] -= c;
    }
    
    for( int i = 1; i <= n; ++i){
        g[i] = g[i] + g[i - 1];
        cout << g[i] << " ";
    }
    return 0;
}

0x03：二维前缀和

【题目】
输入一个n行m列的整数矩阵，再输入q个询问，每个询问包含四个整数x1, y1, x2, y2，表示一个子矩阵的左上角坐标和右下角坐标。
对于每个询问输出子矩阵中所有数的和。

【输入格式】
第一行包含三个整数n，m，q。
接下来n行，每行包含m个整数，表示整数矩阵。
接下来q行，每行包含四个整数x1, y1, x2, y2，表示一组询问。
【输出格式】
共q行，每行输出一个询问的结果。

【数据范围】
1≤n,m≤1000 ,
1≤q≤200000 ,
1≤x1≤x2≤n ,
1≤y1≤y2≤m ,
−1000≤矩阵内元素的值≤1000

【输入样例】

3 4 3
1 7 2 4
3 6 2 8
2 1 2 3
1 1 2 2
2 1 3 4
1 3 3 4

【输出样例】

17
27
21

【解题思路】
二维前缀和
二维前缀和数组：g[i][j] = g[i -1][j] + g[i][j - 1] - g[i - 1][j - 1] + g[i][j]
获取矩阵 （x1, y1）,(x2, y2) (x2 > x1, y2 > y1) 之间的数据和：
sum = g[x2][y2] - g[x1 - 1][y2] - g[x2][y1 - 1] + g[x1 - 1][y1 - 1]

【CODE】

#include<iostream>

using namespace std;

const int N = 1010;

int g[N][N];
int n, m, q;
int x1, x2, y1, y2;

int main(){
    cin >> n >> m >> q;
    for(int i = 1; i <= n; ++i){
        for(int j = 1; j <= m; ++j){
            cin >> g[i][j];
            g[i][j] = g[i][j] + g[i - 1][j] + g[i][j - 1] - g[i - 1][j - 1];
            // cout << g[i][j] << " ";
        }
        // cout << endl;
    }
    
    while(q --){
        cin >> x1 >> y1 >> x2 >> y2;
        
        cout << g[x2][y2] - g[x1 - 1][y2] - g[x2][y1 -1] +g[x1 - 1][y1 - 1] << endl;
    }
    return 0;
}

#include<iostream>

using namespace std;

const int N = 1010;
int a[N][N], b[N][N]; // 原数组， 差分数组

void func(int x1, int y1, int x2, int y2, int c){
    b[x1][y1] += c;  // 所需矩阵加上c
    
    // 右下部分 保持平衡
    b[x2 + 1][y1] -= c;
    b[x1][y2 + 1] -= c;
    b[x2 + 1][y2 + 1] += c;
}

int main(){
    int n, m ,q;
    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)
            func(i, j, i, j, a[i][j]);
    
    // 插入数据        
    for(int i = 0; i < q; ++i){
        int x1, x2, y1, y2, c;
        cin >> x1 >> y1 >> x2 >> y2 >> c;
        func(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] + b[i - 1][j] + b[i][j - 1] -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;
}