前缀和的初步了解与应用

1.一维

前缀和就是从位置1到位置i这个区间内的所有的数字之和。

作用：用来求这个数组中某一段的和

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

#include <bits/stdc++.h>

using namespace std;

const int N = 100010;
int n, m;
int a[N], s[N];

int main() {
    cin >> n >> m;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
    }
//前缀和初始化
    for (int i = 1; i <= n; i++) {//下标从一开始方便处理，a[0]=0;
        s[i] = s[i - 1] + a[i];
    }
//区间和计算
    while (m--) {
        int l, r;
        cin >> l >> r;
        cout << s[r] - s[l - 1] << endl;
    }

    return 0;
}

 

2.二维

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 <bits/stdc++.h>

using namespace std;

const int N = 100;
int n, m;
int a[N][N], s[N][N];

int main() {
    cin>>n>>m;
    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];
        }
    }

    int x1,y1,x2,y2;
    cin>>x1>>y1>>x2>>y2;
    cout<<s[x2][y2]-s[x1-1][y2]-s[x2][y1-1]+s[x1-1][y1-1];
    
    return 0;
}