什么是前缀和
前缀和为一个数组前几个数的和。
比如,定义一个 a a a 数组,和一个前缀和数组 s s s。
那么 s i = ∑ j = 1 i a j s_i=\sum^{i}_{j=1} a_j si=∑j=1iaj,即 a a a 数组前 i i i 个数的和。
同时,我们还可以得出 s i s_i si 的递推公式 s i = s i − 1 + a i , s 0 = 0 s_i=s_{i-1}+a_i,s_0=0 si=si−1+ai,s0=0。
举个例子,设
a
a
a 数组的元素为
1
,
1
,
4
,
5
,
1
,
4
1,1,4,5,1,4
1,1,4,5,1,4(似乎似曾相识),那
s
s
s 数组的元素为
1
,
2
,
6
,
11
,
12
,
16
1,2,6,11,12,16
1,2,6,11,12,16。
前缀和的作用
B3612【深进1.例1】求区间和
题目描述
给定 n n n 个正整数组成的数列 a 1 , a 2 , ⋯ , a n a_1, a_2, \cdots, a_n a1,a2,⋯,an 和 m m m 个区间 [ l i , r i ] [l_i,r_i] [li,ri],分别求这 m m m 个区间的区间和。
输入格式
第一行,为一个正整数 n n n 。
第二行,为 n n n 个正整数 a 1 , a 2 , ⋯ , a n a_1,a_2, \cdots ,a_n a1,a2,⋯,an
第三行,为一个正整数 m m m 。
接下来 m m m 行,每行为两个正整数 l i , r i l_i,r_i li,ri ,满足 1 ≤ l i ≤ r i ≤ n 1\le l_i\le r_i\le n 1≤li≤ri≤n
输出格式
共 m m m 行。
第 i i i 行为第 i i i 组答案的询问。
样例 #1
样例输入 #1
4
4 3 2 1
2
1 4
2 3
样例输出 #1
10
5
说明/提示
样例解释:第 1 到第 4 个数加起来和为 10。第 2 个数到第 3 个数加起来和为 5。
对于
50
%
50\%
50% 的数据:
n
,
m
≤
1000
n,m\le 1000
n,m≤1000 ;
对于
100
%
100\%
100% 的数据:
1
≤
n
,
m
≤
1
0
5
1 \leq n,m\le 10^5
1≤n,m≤105,
1
≤
a
i
≤
1
0
4
1 \leq a_i\le 10^4
1≤ai≤104。
思路讲解
这是一道前缀和的经典例题。
考虑直接暴力枚举,发现时间复杂度为
O
(
n
m
)
O(nm)
O(nm),会 TLE
。
仔细观察,不难发现。如果将前 r r r 个数的总和减去前 l − 1 l-1 l−1 个数的总和,不就是区间 [ l , r ] [l,r] [l,r] 的和了吗?所以区间 [ l , r ] [l,r] [l,r] 的和可以表示为 s r − s l − 1 s_r-s_{l-1} sr−sl−1。
比如,设 a a a 数组的元素为 1 , 1 , 4 , 5 , 1 , 4 1,1,4,5,1,4 1,1,4,5,1,4,那 s s s 数组为 1 , 2 , 6 , 11 , 12 , 16 1,2,6,11,12,16 1,2,6,11,12,16,要想求区间 [ 2 , 4 ] [2,4] [2,4] 的和,只需将 s 4 − s 1 s_4-s_1 s4−s1,答案为 11 − 1 = 10 11-1=10 11−1=10,容易证明是正确的。
这样,我们可以维护一个前缀和数组。对于每一次询问,只需输出 s r − s l − 1 s_r-s_{l-1} sr−sl−1 即可。
算法复杂度为 O ( n + m ) O(n+m) O(n+m)。
代码实现
#include <bits/stdc++.h>
using namespace std;
int m,n,a,s[100001];
int main(){
cin>>n;
for(int i=1;i<=n;i++){
cin>>a;
s[i]=s[i-1]+a;
//一边输入,一边为s[i]赋值
}
cin>>m;
int l,r;
for(int i=1;i<=m;i++){
cin>>l>>r;
cout<<s[r]-s[l-1]<<endl;
}
return 0;
}
所以,相信大家已经看了出来了。前缀和就是用来求区间和的。