【数据结构】前缀和

数据结构-前缀和


前置知识
  • C++入门

思路

区间问题是一种经典问题。
区间问题的一个经典形式是:给定一个长度为 n n n 的数组, q q q 次询问,每次询问输出区间 [ l , r ] [l,r] [l,r] 中的元素和。
这是前缀和的经典例题。
首先考虑暴力。
每次询问,枚举区间,依次求和。
时间复杂度 O ( n + n q ) O(n+nq) O(n+nq),直接爆炸。
天空一声巨响,前缀和闪亮登场
不难发现关于 [ l , r ] [l,r] [l,r] 区间和的一些性质:
∑ i = l r a i = ∑ i = 1 r a i − ∑ i = 1 l − 1 a i \sum_{i=l}^ra_i=\sum_{i=1}^ra_i-\sum_{i=1}^{l-1}a_i i=lrai=i=1raii=1l1ai
而,我们知道
∑ i = 1 k a i = ∑ i = 1 k − 1 a i + a k \sum_{i=1}^ka_i=\sum_{i=1}^{k-1}a_i+a_k i=1kai=i=1k1ai+ak
所以,我们可以得到前缀和数组
s i = { a 1 if  i = 1 s i − 1 + a i if  i > 1 s_i= \begin{cases} a_1&\text{if }i=1\\ s_{i-1}+a_i&\text{if }i>1 \end{cases} si={a1si1+aiif i=1if i>1
那么,对于每一个询问
∑ i = l r a i = s r − s l − 1 \sum_{i=l}^ra_i=s_r-s_{l-1} i=lrai=srsl1


数据结构参数
  • 时间复杂度: Θ ( n + q ) \Theta(n+q) Θ(n+q)
  • 空间复杂度: Θ ( n ) \Theta(n) Θ(n)

实现代码
  • 暴力
    不放了
  • 前缀和
for (int i=1;i<=n;i++)
	cin>>a[i];
for (int i=1;i<=n;i++)
	s[i]=s[i-1]+a[i];
for (int i=1;i<=q;i++){
	int l,r;
	cin>>l>>r;
	cout<<s[r]-s[l-1]<<'\n';
}

练习
  • 洛谷 B3612 求区间和
  • 洛谷 P1444 非常男女
  • 洛谷 P9533 区间翻转区间异或和

你可能感兴趣的:(数据结构,#,前缀和与差分,数据结构)