前缀和

应用

快速求某一范围内的数值的和

方法

一维前缀和

S[i]表示前i个数的和，那么用S[r] - S[l-1] 就可以得到第l个数到第r个数的和（若数组下标从1开始则是[l, r]范围内的数值和）。

二维前缀和

尽量让矩阵横纵坐标从1开始，可以避免很多边界情况。
S[i][j] = S[i-1][j] + S[i][j-1] + arr[i][j] - S[i-1][j-1] 就可以得到以(1,1)为左上角，以(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<vector>
using namespace std;

int main(){
    int n, m;
    cin >> n >> m;
    
    vector<int> arr(n, 0);
    for(auto i = 0; i < n; i++){
        cin >> arr[i];
    }
    
    vector<int> S(n+1, 0);      // S[i] 记录从1到i的和，即前i个数的和
    S[0] = 0;
    for(int i= 0; i < n; i++){
        S[i+1] = arr[i] + S[i];
    }
    for(int i = 0; i < m; i++){
        int l, r;
        cin >> l >> r;
        cout << S[r] - S[l-1] << endl;
    }
    return 0;
}

二维前缀和

#include<iostream>
#include<vector>

using namespace std;

int main(){
    int n,m,q;
    cin >> n >> m >> q;
    vector<vector<int>> arr(n+1, vector<int>(m+1, 0));
    for(int i = 1; i <= n; i++){
        for(int j = 1; j <= m; j++){
            cin >> arr[i][j];
        }
    }
    vector<vector<int>> S(n+1, vector<int>(m+1, 0));        //S[i][j]表示以(1,1)为右上角，以(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] + arr[i][j] - S[i-1][j-1];
        }
    }
    for(int i = 0; i < q; i++){
        int x1, x2, y1, y2;
        cin >> x1 >> y1 >> x2 >> y2;
        int res = S[x2][y2] - S[x1-1][y2] - S[x2][y1-1] + S[x1-1][y1-1];
        cout << res << endl;
    }
    return 0;
}