前缀和算法讲解（一）

今天，我们讲一下前缀和算法。

首先看一道题：https://www.acwing.com/problem/content/797/

我们先写一个暴力的代码，如下：

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

很遗憾，超时了。

我们可以用一个数组s,s[i]表示前i个的和。

我们O(n)的时间复杂度吧s数组预处理，每次查询直接输出s[r]-s[l-1]

画个图：

所以说代码就很好写了：

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

今天讲到这里，下次再见。