【蓝桥杯算法】---第三讲--差分

概念

在引出差分的概念之前，先看如下的两个数组

其差分数组为：

可以观察出a数组其实是b数组的前缀和
a[1]=b[1]
a[2]=b[1]+b[2]
a[3]=b[1]+b[2]+b[3]
a[4]=b[1]+b[2]+b[3]+b[4]…
a[i]=b[1]+b[2]+b[3]…+b[i-1]+b[i]
满足这样关系的数组，我们就称作差分数组，由此也可以看出差分其实是前缀和的逆运算。

一维数组的差分

在某种场景下，题目可能会要求你让一个数组的任意一个区间+上一个数字，这时如果用到差分的思想，就可以做到快速求解。

差分数组的本质：

应用一：如何构造一个差分数组：
核心公式：b[l]+= c; b[r+1] -= c;
代码：

#include<iostream>
using namespace std;
const int N=100010;
int a[N],b[N];
int n,m;

void insert(int l,int r,int c ){
    b[l]+=c;
    b[r+1]-=c;
}


int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    scanf("%d",&a[i]);
    
    for(int i=1;i<=n;i++)
    insert(i,i,a[i]);
 }

应用二：通过差分数组对原数组的任意区间数据进行修改
假设需要修改原数组l-r这个区间的数据
一共两个步骤
步骤1： 通过公式对差分数组进行修改
公式：b[l]+=c ; b[r+1]-=c;
步骤2： 通过修改以后的差分数组求出新的原数组
推导过程：

#include<iostream>
using namespace std;
const int N=100010;
int a[N],b[N];
int n,m;

void insert(int l,int r,int c ){
    b[l]+=c;
    b[r+1]-=c;
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    scanf("%d",&a[i]);
    
    for(int i=1;i<=n;i++)
    insert(i,i,a[i]);//(1)
    
    while(m--){
        int l,r,c;
        scanf("%d%d%d",&l,&r,&c);
        insert(l,r,c);//(2)
    }
    for(int i=1;i<=n;i++)//(3)
    b[i]=b[i-1]+b[i];
    
    for(int i=1;i<=n;i++)
    printf("%d ",b[i]);
}

(1)构造原数组的差分数组
(2)对差分数组进行数据修改
(3)通过差分数组得出新修改后的原数组

二维数组的差分

应用一：如何构造二位差分数组

主要公式：
b[x1][y1] += c;
b[x2 + 1][y1] -= c;
b[x1][y2 + 1] -= c;
b[x2 + 1][y2 + 1] += c;

完整代码：

#include <iostream>
using namespace std;
const int N = 1010;

int n, m, q;
int a[N][N], b[N][N];

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()
{
    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]);

    for (int i = 1; i <= n; i ++ )
        for (int j = 1; j <= m; j ++ )
            insert(i, j, i, j, a[i][j]);
}

应用二：如何通过二维差分数组对原数组任意区间的数据进行修改
假设需要对原数组(x1,y1) (左上角的坐标) ,(x2,y2) (右下角的坐标) 这两个区间的数据进行修改。
一共两个步骤：

步骤1.代入公式:构造原数组的差分数组
b[x1][y1] += c;
b[x2 + 1][y1] -= c;
b[x1][y2 + 1] -= c;
b[x2 + 1][y2 + 1] += c;

推导过程：

(1)公式1使得橙色区域全都加上了c，但是所要求的区间并没有这么大，所以接下来的公式则是对这个区间的修正
(2)公式2使得绿色的区域全都减去了c
(3)公式3同样使得绿色的区域全都减去了c
(4)因为公式3和4对同一共区域都减去了c，所以需要对重叠的部分加回多减去的c，也就是橙色的部分

步骤2.通过差分数组求出新的原数组
这步就用到了二维数组的前缀和公式
b[i][j]=b[i-1][j]+b[i][j-1]-b[i-1][j-1]+b[i][j];

#include<iostream>
using namespace std;
const int N=2001;
int a[N][N],b[N][N];
int n,m,q;

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(){
    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]);
    
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        insert(i,j,i,j,a[i][j]);
        
    while(q--){
        int x1,x2,y1,y2,c;
        scanf("%d%d%d%d%d",&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++)
        b[i][j]=b[i-1][j]+b[i][j-1]-b[i-1][j-1]+b[i][j];
        
        
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++)
        printf("%d ",b[i][j]);
        printf("\n");
    
    }
}

(1)构造原数组的差分数组
(2)对差分数组进行数据修改
(3)通过差分数组得出新修改后的原数组