【数据结构】前缀和

OIer-zyh

已于 2023-09-30 12:33:33 修改

阅读量139

点赞数 5

分类专栏：数据结构 # 前缀和与差分文章标签：数据结构

于 2023-09-30 08:32:49 首次发布

本文链接：https://blog.csdn.net/zhouyh2011/article/details/133426563

版权

数据结构同时被 2 个专栏收录

12 篇文章 0 订阅

订阅专栏

前缀和与差分

5 篇文章 0 订阅

订阅专栏

数据结构-前缀和

前置知识

C++入门

思路

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

数据结构参数

时间复杂度： $\Theta(n+q)$
空间复杂度： $\Theta(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';
}