AcWing795.前缀和

具体做法

类比于数学中的数列求和，原数列a₁,a₂,…a_n，则前缀和S_i = a₁ + a₂ +…+ a_i。

求前缀和核心运算

const int N=100010;
int s[N],a[N];
for(int i=1;i<=n;i++)
    s[i]=s[i-1]+a[i];

查询操作

对于每次查询，只需执行sum[r]-sum[l-1] ，时间复杂度为O(1)

scanf("%d%d",&l,&r);
printf("%d",s[r]-s[l-1]);

原理

s[r] = a[1] + a[2] + … + a[l-1] + a[l] + … + a[r];
s[l-1] = a[1] + a[2] + … + a[l-1];
s[r] - sum[l-1] = a[l] + … + a[r];

这样，对于每个询问，只需要执行 s[r]-s[l-1]。输出原序列中从第l个数到第r个数的和的时间复杂度变成了O(1)。
我们把它叫做一维前缀和。

代码

#include<iostream>
using namespace std;
const int N = 100010;
int n,m;
int s[N],a[N];

int main()
{
	scanf("%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		s[i] = s[i-1] + a[i];//步骤1：计算前缀和
	}
	while(m--)
	{
		int l,r;
		scanf("%d%d",&l,&r);
		printf("%d\n",s[r]-s[l-1]);//步骤2：查询具体区间的前缀和并打印输出
	}
	return 0;
}

总结

做题只需牢记两个步骤：

1. 预处理，求前缀和
   s[i] = s[i-1] + a[i];
2. 查询数组特定部分前缀和，如输入l，r，查询区间[l,r]之间的前缀和
   a[l] + … + a[r] = s[r] - s[l-1];