从零开始的异世界算法*4（前缀和与差分）

前缀和与差分的概念和数列的前n项相似，可用于优化时间复杂度，快速计算一个区间的长度

首先是一维数组的前缀和，求出数组l到r区间上的和，n次读入

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int q[N],a[N];
int n,m;
int main()
{
    cin>>n>>m;
    for(int i=1;i<n;i++)scanf("%d",&q[i]);
    for(int i=1;i<n;i++)a[i]=a[i-1]+q[i];
    while(m--)
    {
        int l,r;cin>>l>>r;
        cout<<a[r]-a[l-1]<<endl;
    }
}

然后是一维数组的差分，可以应用于在某一段连续区间的修改

如下题

输入一个长度为 n的整数序列。

接下来输入 m 个操作，每个操作包含三个整数 l,r,c，表示将序列中 [l,r]之间的每个数加上 c。

请你输出进行完所有操作后的序列

#include<bits/stdc++.h>
using namespace std;
const int N=100010;
int q[N];
int a[N];
int n,m;
void insert(int l,int r,int c)
{
    a[l]+=c;//使得该点后每一个元素加c
    a[r+1]-=c;//使得该点后每一个元素减c
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)scanf("%d",&q[i]);
    for(int i=1;i<=n;i++)insert(i,i,q[i]);//a[i]=q[i]-q[i-1];
    while(m--)
    {
        int l,r,c;
        scanf("%d%d%d",&l,&r,&c);
        insert(l,r,c);
    }
    for(int i=1;i<=n;i++)
    {
        q[i]=q[i-1]+a[i];
        printf("%d ",q[i]);
    }
}

然后是相对复杂的二维数组的前缀和与差分

首先可以通过递推计算的出二维数组前缀和的公式

s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j];

然后计算x1y1x2y2构成的子矩阵的和

#include<bits/stdc++.h>
using namespace std;
const int N=1010;
int a[N][N];
int s[N][N];
int n,m,x;
int main()
{
    cin>>n>>m>>x;
    for(int i=1;i<=n;i++)for(int j=1;j<=m;j++)scanf("%d",&a[i][j]);
    for(int i=1;i<=n;i++)for(int j=1;j<=m;j++)
    s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j];
    while(x--)
    {
        int x1,y1,x2,y2;
        cin>>x1>>y1>>x2>>y2;
        cout<<s[x2][y2]-s[x2][y1-1]-s[x1-1][y2]+s[x1-1][y1-1]<<endl;
    }
}

再看看差分，先构造一个b数组使得b数组的前缀和为a数组

#include<bits/stdc++.h>
using namespace std;
const int N=1010;
int a[N][N];
int s[N][N];
int n,m,k;
void insert(int x1,int y1,int x2,int y2,int c)
{
    a[x1][y1]+=c;
    a[x2+1][y1]-=c;
    a[x1][y2+1]-=c;
    a[x2+1][y2+1]+=c;
}
int main()
{
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)scanf("%d",&s[i][j]);
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            insert(i,j,i,j,s[i][j]);
        }
    }
    while(k--)
    {
        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++)
        {
          // a[i][j]=a[i][j]+a[i-1][j]+a[i][j-1]-a[i-1][j-1];
           //printf("%d ",a[i][j]);
           s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j];
           printf("%d ",s[i][j]);
        }
        printf("\n");
    }
}