前缀和（模板+习题）C/C++

前缀和概述       

        前缀和就是从位置1到位置i这个区间内的所有的数字之和。我们用前缀和有一个很大的优势，就是可以快速的得到某一个区间的区间总和
        对于每个询问，只需要执行 sum[r] - sum[l - 1]。输出原序列中从第l个数到第r个数的和的时间复杂度变成了O(1)。我们把它叫做一维前缀和。
        公式：s[r]-s[l - 1] = a[l] + a[l + 1] + ……+a[r]
        二维前缀和预处理公式: s[i][j] = s[i - 1][j] + s[i][j - 1] + a[i][j] - s[i - 1][j - 1]

例题

        题目描述   

输入一个长度为 n 的整数序列。接下来再输入 m 个询问，每个询问输入一对 l,r。 

对于每个询问，输出原序列中从第 l 个数到第 r 个数的总和。 

输入格式： 第一行包含两个整数 n 和 m。 第二行包含 n 个整数，表示整数数列。 接下来 m 行，每行包含两个整数 l 和 r，表示一个询问的区间范围。 

输出格式 共 m 行，每行输出一个询问的结果。 数据范围 1≤l≤r≤n 1≤n,m≤100000, −1000≤数列中元素的值≤1000

          样例输入

5 3
2 1 3 6 4
1 2
1 3
2 4

           样例输出

3
6
10

           AC代码

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int a[N], sum[N], n, m;
int main () {
    cin >> n >> m;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        sum[i] = sum[i - 1] + a[i];
    }
    int l, r;
    while (m--) {
        cin >> l >> r;
        cout << sum[r] - sum[l - 1] << endl;
    }
    return 0;
}

还有二维的……

题目描述

输入一个长度为 n 的整数序列。接下来再输入 m 个询问，每个询问输入一对 l,r。 

对于每个询问，输出原序列中从第 l 个数到第 r 个数的总和。 

输入格式： 第一行包含两个整数 n 和 m。 第二行包含 n 个整数，表示整数数列。 接下来 m 行，每行包含两个整数 l 和 r，表示一个询问的区间范围。 

输出格式 共 m 行，每行输出一个询问的结果。 数据范围 1≤l≤r≤n 1≤n,m≤100000, −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

AC代码

#include <bits/stdc++.h>
using namespace std;
const int N = 1010;
int n, m, q, x1, x2, y1, y2, a[N][N], sum[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];
            sum[i][j] = sum[i - 1][j] + sum[i][j - 1] - sum[i - 1][j - 1] + a[i][j];
        }
    }
    while (q--) {
        cin >> x1 >> y1 >> x2 >> y2;
        cout << sum[x2][y2] - sum[x1 - 1][y2] - sum[x2][y1 - 1] + sum[x1 - 1][y1 - 1]<<endl;
    }
    return 0;
}

习题

最大子段和 - 洛谷

#include <bits/stdc++.h>
using namespace std;
const int N = 2e5 + 10;
int temp, value = -22222, sum, maxn = -2222222;
int m;
bool st;
int main () {
    cin >> m;
    while (m--) {
        cin >> temp;
        if (temp >= 0) {
            st = true;
        }
        sum += temp;
        maxn = max(temp, maxn);
        if (sum < 0) {
            sum = 0;
        }
        value = max(value, sum);
    }
    if (!st) {
        cout << maxn;
    } else {
        cout << value;
    }
    return 0;
}

[HNOI2003] 激光炸弹 - 洛谷

#include <bits/stdc++.h>
using namespace std;
const int N = 5010;
int n, m;
int s[N][N];
int main () {
    int cnt, r;
    cin >> cnt >> r;
    r = min(5001, r);
    n = m = r;
    while (cnt--) {
        int x, y, w;
        cin >> x >> y >> w;
        x++, y++;
        n = max(n, x);
        m = max(m, y);
        s[x][y] += w;
    }
    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];
        }
    }
    int res = 0;
    for (int i = r; i <= n; i++) {
        for (int j = r; j <= m; j++) {
            res = max(res, s[i][j] - s[i - r][j] - s[i][j - r] + s[i - r][j - r]);
        }
    }
    cout << res << endl;
    return 0;
}