算法介绍
前缀和是为了处理在数组中,进行多次区间总和查询并输出查询结果的基本算法。
简单的来说,如果在遇到前缀和问题时,不使用前缀和处理,每次查询需要执行查询范围数据个数次的加法,当查询次数过多,或者查询范围较大时,就需要执行多次加法,当使用暴力解法时会出现时间超限的情况。
一维前缀运算的减少
如果使用了前缀和算法,在所有查询之前,进行一个n次的加法运算(即前缀和预处理),之后的每次查询,仅需要执行一次减法运算。
具体做法:
首先做一个预处理,定义一个sum[]数组,sum[i]代表a数组中前i个数的和。
预处理(求前缀和运算):
前缀和的预处理中sum[i]=sum[i-1]+a[i],因为在算sum[i]时i-1的前缀和及sum[i-1]已经求出再加上a[i]就是i的前缀和sum[i]了。不过要注意的一点是,前缀和通常会开辟n+1大小的数组,并使sum[0]=0来解决数组超限问题。
const int N=1e5+10;
int sum[N],a[N]; //sum[i]=a[1]+a[2]+a[3].....a[i];
for(int i=1;i<=n;i++)
{
sum[i]=sum[i-1]+a[i];
}
然后查询操作:
scanf("%d%d",&l,&r);
printf("%d\n", sum[r]-sum[l-1]);
对于每次查询,只需执行sum[r]-sum[l-1] ,时间复杂度为O(1)。
题目链接
https://www.acwing.com/problem/content/797/
题目描述
输入一个长度为 n 的整数序列。
接下来再输入 m 个询问,每个询问输入一对 l,r。
对于每个询问,输出原序列中从第 ll 个数到第 rr 个数的和。
输入格式
第一行包含两个整数 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
#include<stdio.h>
const int N=100010;
int k[N];
int s[N];
int main()
{
int n, m;
scanf("%d %d", &n, &m);
for (int i = 1; i <=n; i++)scanf("%d", &k[i]);
//前缀和预处理
for (int i = 1; i <=n + 1; i++) s[i] = s[i - 1] + k[i];
while (m--)
{
int l, r;
scanf("%d %d", &l, &r);
//输出查询结果
printf("%d\n", s[r] - s[l - 1]);
}
return 0;
}