【算法入门】前缀和与差分

前言

本文将带你一起了解前缀和与差分这个算法小知识，阐明其原理以及相关的例题，再给出代码模板，希望对大家有所帮助~

文章目录

一维前缀和

定义

S[i] = a[1] + a[2] + … a[i]
a[l] + … + a[r] = S[r] - S[l - 1]
根据这个公式我们可以看到，如果要计算数组中，区间[l,r]上的元素之和，就就不用去遍历数组了，只需要利用S[r] - S[l - 1]就可以得到结果啦！

例题实践

#include<iostream>
using namespace std;
const int N=100010;
int a[N];
int l,r,n,m;
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        a[i]=a[i-1]+a[i];//求前缀和
    }
    while(m--)
    {
        cin>>l>>r;
        cout<<a[r]-a[l-1]<<endl;//利用公式算出【l,r】上的和；
    }
    return 0;
}

一维差分

定义

差分实际上就是前缀和的逆运算，假设两个数组a[N]与b[N],如果数组a[N]是b[N]的前缀和，那么b数组就是a的差分。也就是有：
a[i]=b[1]+b[2]+…+b[i];那这有什么用?如果我们需要给a数组的区间【l,r】上每个元素都加上一个常数c,想想我们可以怎么做？我们当然可以遍历数组a,逐个加上c,但是有了差分数组b之后，我们还可以这么做，让b[l]+=c,b[r+1]-=c;为什么可以这样做？请看图解：

如果给b数组在 l 这个位置上加上c的话，那么b的前缀和是数在 l 位置之后每隔元素都加上了c,但我们希望，只是在【l,r】上加c,所以在 r 之后减去c 就可以了.那现在的问题就是，怎么去得到这个b数组？我们先假设b数组全是0,a数组也全是0,但实际上，a数组并不是0，既然在b数组上进行b[l]+c,b[r+1]-c，相当于在a数组上给每个元素加上c,那么我在区间[x,x]上进行这个操作，是不是就相当于把a[x]这个元素加上c;

例题实践

#include<iostream>
using namespace std;
const int N=100010;
int a[N],b[N];
int n, m;
int l,r,c;
void insert(int l,int r,int c)
{
    b[l]+=c;
    b[r+1]-=c;
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        insert(i,i,a[i]);//得到b数组（a的差分）
    }
    while(m--)
    {
        cin>>l>>r>>c;
        insert(l,r,c);
    }
    for(int i=1;i<=n;i++) a[i]=a[i-1]+b[i];//得到加完之后的a数组（b的前缀和）
    for(int i=1;i<=n;i++) cout<<a[i]<<' ';
}

二维前缀和

定义

S[i, j] = 第i行j列格子左上部分所有元素的和
以(x1, y1)为左上角，(x2, y2)为右下角的子矩阵的和为：
S[x2, y2] - S[x1 - 1, y2] - S[x2, y1 - 1] + S[x1 - 1, y1 - 1]

s[i][j]的值就是下面红色的元素之和：

s[x1][y1]的是：

s[x2][y2]的内容是：

要求的是黄色这部分的和：

蓝色部分是s[x2][y1-1]

紫色部分是s[x2-1][y2]

所以s[x1-1][y1-1]被减了两遍，要加回来一次

怎么样现在理解这个公式是怎么来的了吧？在看例题;

例题实践

#include<iostream>
using namespace std;
const int N=1010;
int n,m,q;
int a[N][N],s[N][N];
int main()
{
    scanf("%d %d %d",&n,&m,&q);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        {
            scanf("%d",&a[i][j]);
            s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j];//计算二维前缀和。
        }
        int x1,y1,x2,y2;
        while(q--)
        {
            scanf("%d %d %d %d",&x1,&y1,&x2,&y2);
            cout<<s[x2][y2]-s[x2][y1-1]-s[x1-1][y2]+s[x1-1][y1-1]<<endl;
        }
    return 0;
}

二维差分

定义

二维差分也就是二维前缀和的逆运算，a是b的前缀合，b就是a的差分，作用和一维的类似，如果我们要给一个给以(x1, y1)为左上角，(x2, y2)为右下角的子矩阵每个元素加上c.只需要在差分矩阵上做以下操作：
给以(x1, y1)为左上角，(x2, y2)为右下角的子矩阵中的所有元素加上c：
S[x1, y1] += c, S[x2 + 1, y1] -= c, S[x1, y2 + 1] -= c, S[x2 + 1, y2 + 1] += c

要给红色区域每个元素都加上c,考虑先给它的差分矩阵上x1,y1出加上c

但问题是，粉红色区域也都会加上C

那让（x2+1,y1）这个点减上之后：

还有一部分没有处理到，再考虑让（x1,y2+1）这个点减C:

问题是黑色这部分又减了两次C，所以要加回来，应该在(x2+1,y2+1)上加上C

这样操作之后正好是给我们的目标区域全部加上C；

例题实践

#define _CRT_SECURE_NO_WARNINGS 1
#include<iostream>
using namespace std;
const int N = 1010;
int n, m,q;
int a[N][N], b[N][N];
int x1, y1, x2, y2,c;
void insert(int x1, int y1, int x2, int y2,int c)
{
	b[x1][y1] += c;
	b[x2 + 1][y1] -= c;
	b[x1][y2 + 1] -= c;
	b[x2 + 1][y2 + 1] += c;

}
int main()
{
	cin >> n >> m >> q;
	for(int i=1;i<=n;i++)
		for (int j = 1; j <= m; j++)
		{
			scanf("%d", &a[i][j]);
			insert(i, j, i, j, a[i][j]);
		}
	while (q--)
	{
		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 - 1][j] + a[i][j - 1] - a[i - 1][j - 1] + b[i][j];
			cout << a[i][j] << ' ';
		}
		cout << endl;
	}
}