前缀和与差分

一.前缀和

假如一个数组中有n个数，a[1]、a[2]、a[3]…a[i]…a[n]。要求数组a[1]到a[n]的和，正常情况下，我们要用一个循环编历这个数组，来求这个数组的和，这样的话时间复杂度为o(n)，而如果用前缀和进行优化的话，就可以将时间复杂度从o(n)直接优化到o(1)，是一个非常常用的数组求和的优化方法。
一维前缀和就是将前面的数进行累加s[n]=a[1]+a[2]+...+a[n],s[n]就相当于a[1]到a[n]的和。
看一下代码：

int n,a[100];
cin>>n;
a[0]=0;               //初始化，将a[0]赋成0
for(int i=1;i<=n;i++)
{
     cin>>a[i];       //输入n个数
     a[i]+=a[i-1];    //前缀和处理
}
//如果要求数组从1到n的和
cout<<a[n]<<endl;
//求从i到j(i<=j)的和，只需要让a[j]减去a[i-1]就可以了
cout<<a[j]-a[i-1]<<endl;

以上是一维前缀和，现在我们来把一位的前缀和推广到二维。二维前缀和与一维基本类似，但比一维前缀和稍微复杂了一些。
二维前缀和就是求两个维度上的和，s[n][m]=a[1][1]+a[1][2]+....+a[n][m]s[n][m]就相当于a[1][1]到a[n][m]的和。
看一下代码

int n,m,a[100][100]={0};     //将a[][]数组清0
cin>>n>>m;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
     cin>>a[i][j];
     a[i][j]+=a[i-1][j]+a[i][j-1]-a[i-1][j-1];//让a[i][j]加a[i-1][j]和a[i][j-1]
}                 //因为a[i-1][j]和a[i][j-1]都包含了a[i-1][j-1],所以a[i][j]要
                  //减去一个a[i-1][j-1]
//求(1,1)到(n,m)的和
cout<<a[n][m]<<endl;
//求(x1,y1)到(x2,y2)(x2>=x1&&y2>=y1)的和
cout<<a[x2][y2]-a[x2][y1-1]-a[x1-1][y2]+a[x1-1][y1-1]<<endl;
//因为a[x2][y2]-a[x2][y1-1]-a[x1-1][y2]会多剪掉一个a[x1-1][y1-1]，所以在后面我们
//要加回来

如果还是不懂的可以自己画一下图看看，在这里我就不画了。

二.差分

差分没有前缀和那么常用，但也是必须要掌握的。差分可以说是前缀和的逆运算，前缀和是每一位都加上前一位，而差分则是每一位都减去前一位,即d[n]=a[n]-a[n-1].

差分的作用就是:假设d[]是数组a[]的差分，如果想把a[]中的数全部加上x，那么就需要编历数组a，每个数加x，这样时间复杂度位o(n)，但如果用差分来修改，就只需要o(1)的复杂度即可完成这个操作。
我们来看一个题：
输入一个长度为n的整数序列。
接下来输入m个操作，每个操作包含三个整数l, r, c，表示将序列中[l, r]之间的每个数加上c。最后输出修改完的序列。
代码如下：

#include <iostream>
using namespace std;
int d[100];
void insert(int l,int r,int c)   //插入函数
{                                //可以用它来向区间l到r上插入c
     d[l]+=c;                    //也可以用它插入数据
     d[r+1]-=c;
}
int main()
{
     int n,m;
     cin>>n>>m;
     for(int i=1;i<=n;i++)       //输入n个数，并将其插入数组d中
     {
          int c;
          cin>>c;
          insert(i,i,c);
     }
     while(m--)                  //利用差分，将区间l到r上都加上c
     {
          int l,r,c;
          cin>>l>>r>>c;
          insert(l,r,c);
     }
     for(int i=1;i<=n;i++)       //用前缀和回复原来的数组并输出
     {
          d[i]+=d[i-1];
          cout<<d[i]<<" ";
     }
     return 0;
}

以上是一维的差分，现在我们也把它推广到二维。
二维的差分和二维的前缀和也差不多，只要理解了二维的前缀和，那二维的差分也没不难理解。

我们再来看一个题：
输入一个n行m列的整数矩阵，再输入q个操作，每个操作包含五个整数x1, y1, x2, y2, c，其中(x1, y1)和(x2, y2)表示一个子矩阵的左上角坐标和右下角坐标。
每个操作都要将选中的子矩阵中的每个元素的值加上c。
最后输出操作完的矩阵。
代码如下：

#include <iostream>
using namespace std;
int d[100][100];
void insert(int x1,int y1,int x2,int y2,int c)
{                         //插入函数，可以用它来插入数据
    d[x1][y1]+=c;         //也可以用它来将(x1,y1)到(x2,y2)上加上c
    d[x1][y2+1]-=c;
    d[x2+1][y1]-=c;
    d[x2+1][y2+1]+=c;
}
int main()
{
    int n,m,q;
    cin>>n>>m>>q;
    for(int i=1;i<=n;i++) //输入数据，并将其插入数组
    for(int j=1;j<=m;j++)
    {
        int x;
        cin>>x;
        insert(i,j,i,j,x);
    }
    while(q--)           //利用差分，将(x1,y1)到(x2,y2)上加上c
    {
        int x1,y1,x2,y2,c;
        cin>>x1>>y1>>x2>>y2>>c;
        insert(x1,y1,x2,y2,c);
    }
    //利用前缀和来恢复原数组并输出
    for(int i=1;i<=n;i++){
    for(int j=1;j<=m;j++){
             d[i][j]+=d[i-1][j]+d[i][j-1]-d[i-1][j-1];
             cout<<d[i][j]<<" ";
        }
        cout<<endl;
    }
    return 0;
}

相关习题：
Sweets Eating