前缀和算法和差分算法

我们在学习算法的过程中经常会听到前缀和和差分算法，这里我直接向大家介绍前缀和和差分算法的内容和用法。你可以直接死记硬背，不用理解原理。

前缀和

假设我们有一个数组arr[n]，我们询问m次 对任意给定的l和r，从arr[l]一直加到arr[r]的和是多少。

我们可以直接使用暴力算法

int l,r;
int sum;
int m;
cin>>m;
while(m--){
    cin>>l>>r;
    sum=0;
    for(int i=l;i<=r;i++)
    {
        sum+=arr[i];
    }
    
    cout<<sum;
}

这样sum就是从arr[i]加到arr[r]的和了。但是这样的话时间复杂度可能会很大，所以我们学习前缀和算法，降低时间复杂度。

下面我们直接上前缀和算法的代码

int sum[n];
sum[0]=arr[0];
for(int i=1;i<n;i++)
{
    sum[i]=sum[i-1]+arr[i];
}

我们先设了一个数组sum[n]，然后让sum[0]，也就是sum[]数组的第一项等于arr[]数组的的一项，然后for循环，从i=1开始，令sum[i]=sum[i-1]+arr[i]。

这样，得到的sum[i]就是arr[]数组从  arr[0]  加到  arr[i]  的和了。

所以此时，我们想求从arr[l]加到arr[r]的和就简单了，这个和就是sum[r]-sum[l-1]。

所以我们利用前缀和算法解决上述问题的代码就是

int sum[n];
sum[0]=arr[0];
for(int i=1;i<n;i++)
{
    sum[i]=sum[i-1]+arr[i];
}

int m,l,r;
cin>>m;

while(m--){
    cin>>l>>r;
    cout<<sum[r]-sum[l-1];
}

这样时间复杂度就会大大降低。

差分

学完前缀和，我们来学差分。

假设我们要进行m次操作，每次操作都是把arr[]数组中从下标为l的项到下标为r的项都加上一个数c（c可以是正也可以是负）那么我们可以直接暴力写

int c;
cin>>c;

int m,l,r;
cin>>m;
while(m--)
{
    cin>>l>>r;
    for(int i=l;i<=r;i++)
    {
        arr[i]+=c;
    }
}

同样的，这样可能会有一个问题，时间复杂度太大，直接超时。所以我们用差分算法，来求m次操作后的arr[]数组。

下面我们开看代码。

int d[n];
d[0]=arr[0];
for(int i=1;i<n;i++)
{
    d[i]=arr[i]-d[i-1];
}

我们先设了一个数组d[]，然后让d[0]=arr[0]，然后for循环，从i=1开始，令d[i]=arr[i]-d[i-1]，这样我们得到数组d[n]

d[l]=d[l]+c;
d[r+1]=d[r+1]-c;

int s[n];
s[0]=d[0];
for(int i=1;i<n;i++)
{
    s[i]=s[i-1]+d[i];    
}

然后我们d[l]=d[l]+c，d[r+1]=d[r+1]-c。我们再设一个数组s[n]，令s[0]=d[0],然后for循环，从i=1开始，令s[i]=s[i-1]+d[i]，这样得到了数组s[n]

数组s[n]就是经过一次操作后的arr[n]。

所以求经过m次操作后的arr[]数组，其代码就是

int d[n];
d[0]=arr[0];
for(int i=1;i<n;i++)
    {
        d[i]=arr[i]-d[i-1];
    }

while(m--){

    cin>>l>>r;

    d[l]=d[l]+c;
    d[r+1]=d[r+1]-c;
}

int s[n];
s[0]=d[0];
for(int i=1;i<n;i++)    
{
     s[i]=s[i-1]+d[i];
}

这样我们得到的s[n]就是我们要求的改变后的arr[n]。