前缀和维护可以是数列,也可以是矩阵。
以一个小题为例:给定n个数ai以及m个询问并每次询问一段区间的和。
要求:一个O(n+m)的做法。
要求效率如此之高,看来,强大的树状数组和线段树也无能为力了。
那么如此一来,前缀和上场了。若想时间复杂度控制在O(n+m),光读一遍就需要O(n),即
每一个询问要求效率为O(1),可怕~~
但前缀和可以做到,开两个数组a[n]和s[n],前者装数据,后者装前n项和。
若想求一段区间[l,r]的和,只需用s[r]-s[l],即可解决。
#include
using namespace std;
int main()
{
scanf("%d",&n);
for (i=1; i<=n; i++) scanf("%d",&a[i]);
for (i=1; i<=n; i++) s[i]=s[i-1]+a[i];
scanf("%d",&m);
for (i=1; i<=m; i++)
{
scanf("%d%d",&A,&B);