第十章：前缀和与差分

 前缀和

例题1：前缀和

给定一个长度为N的数组：a1,a2,a3......an。（N<=10^6，1<= A[i]<=10^3）。接下来有M次查找，每次查找给两个整数l和r。求a[l]到a[r]的子段和是多少。（M<=10^6）。

解法1：直接枚举法（时间复杂度为n*n）

#include<iostream>
using namespace std;
int a[100005],n,m;
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    cin>>m;
    for(int i=1;i<=m;i++)
    { 
        int l,r,sum=0; 
        cin>>l>>r;
        for(int j=l;j<=r;j++)
            sum+=a[j]; 
        cout<<sum<<endl; 
    }
    return 0;
} 

Copy

解法2：利用前缀和求解（时间复杂度为n）

前缀和也称前缀和数组：给定一个数组A，定义数组sum，数组中每一项sum[i]保存数组 A[i]到A[j]的和，数组sum称为数组A的前缀和数组。

如何计算前缀和？

for(int i=1;i<=n;i++)

    for(int j=1;j<=i;j++)

        sum[i]=sum[i]+a[j];

Copy

sum[5]=A[1]+A[2]+A[3]+A[4]+A[5];
sum[5]=sum[4]+A[5];

Copy

通项公式：sum[i]=sum[i-1]+A[i];

举个栗子：设l=4, r=6;则子段A[4,6]=A[4]+A[5]+A[6]=sum[6]-sum[3];

sum[6]=A[1]+A[2]+A[3]+A[4]+A[5]+A[6];
sum[3]=A[1]+A[2]+A[3];

Copy

通项公式：A[l,r]=sum[r]-sum[l-1];

详细代码如下：

#include<iostream>
using namespace std;
int a[100005],sum[100005],n,m;
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        sum[i]=sum[i-1]+a[i];  //计算前缀和
    }  
    int l,r; 
    for(int i=1;i<=m;i++)
    {
        cin>>l>>r;
        cout<<sum[r]-sum[l-1]<<endl;  //计算子段和A[l,r] 
    }