4.差分算法

差分就是前缀和的逆运算。跟前缀和类似,给定一些数组a1,a2....an,也给定一些数组b1,b2...bn,其中a[i]=b[i]+a[i-1],也就是说ai是bi的前缀和,反过来bi就是ai的差分数组了。

所以:b[i] = a[i] - a[i - 1]。

题目:

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

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

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

输入格式

第一行包含两个整数 nm

第二行包含 n 个整数,表示整数序列。

接下来 m行,每行包含三个整数 lrc,表示一个操作。

输出格式

共一行,包含 n个整数,表示最终序列。

数据范围

1≤n,m≤100000

1≤l≤r≤n

−1000≤c≤1000

−1000≤整数序列中元素的值≤1000

输入样例:
6 3
1 2 2 1 2 1
1 3 1
3 5 1
1 6 1
输出样例:
3 4 5 3 4 2

题目分析,相比较前缀和题目,差分题目中是要求从l到r中的数组加r的和。始终要记得,a数组是b数组的前缀和,要求ai+c从l到r的和,也就是把b数组也同时加上r,但是是从b[l]开始加的,一直加到b[r],所以b[r+1]之后的数组要变成b[r+1]-c。画个图说明一下:

题目中求从l到r的a数组加上c的和,所以我们在b数组从b[l]开始全部加上r一直加到最后,因为a[i]=b[i]+a[i-1],也就是说b[i] = a[i] - a[i - 1],所以在b[r+1]之后的元素要减去c,减去c的元素就是原来普通的b元素的值。

因此我们得出一维差分结论:给a数组中的[ l, r]区间中的每一个数都加上c,只需对差分数组b做

b[l] + = c, b[r+1] - = c

第一步,创建差分数组b[i]=a[i]-a[i-1],第二步,b[l] + = c, b[r+1] - = c,第三步,a[i]=b[i]+a[i-1].

代码实现:

#include<iostream>
using namespace std;
const int N=100010;
int a[N],b[N];
int main(){
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        b[i]=a[i]-a[i-1];//第一步,创建差分数组b[i]=a[i]-a[i-1]
    }
    int l,r,c;
    while(m--){
        cin>>l>>r>>c;
        b[l]+=c;
        b[r+1]-=c;//第二步,写出b[l]和b[r+1]表达式
    }
    for(int i=1;i<=n;i++){
       a[i] = b[i] + a[i - 1]; //第三步,写出a[i]的表达式
       cout<<a[i]<<" ";
    }
    return 0;
}

二维差分:

跟一维差分一样,给出了a[ij]数组,其中b[ij]数组为a[ij]数组的差分数组,a[ij]数组为b[ij]数组的前缀和数组。类比一维差分,我们给出了以下操作使之成为二维差分:

b[x1][y1] + = c;相当于从左上角边界一直到右下角的整个矩形都加上c

b[x1,][y2+1] - = c;相当于图中绿色的部分全部减去c,得到原本的b[ij]的值

b[x2+1][y1] - = c;相当于图中红色的部分全部减去c,得到原本的b[ij]的值

b[x2+1][y2+1] + = c;从图中可以看到,一开始从左边界到右下角的数组元素全部被加上c,但是被红色和绿色都剪去了c,也就是加了一个c减去了两个c,所以还要加上一个c

我们将上述操作构造一个insert插入函数:

void insert(int x1,int y1,int x2,int y2,int c)
{     //对b数组执行插入操作,等价于对a数组中的(x1,y1)到(x2,y2)之间的元素都加上了c
    b[x1][y1]+=c;
    b[x2+1][y1]-=c;
    b[x1][y2+1]-=c;
    b[x2+1][y2+1]+=c;
}

我们可以先假想a数组为空,那么b数组一开始也为空,但是实际上a数组并不为空,因此我们每次让b数组以(i,j)为左上角到以(i,j)为右下角面积内元素(其实就是一个小方格的面积)去插入 c=a[i][j],等价于原数组a中(i,j) 到(i,j)范围内 加上了 a[i][j] ,因此执行n*m次插入操作,就成功构建了差分b数组。

代码如下:

for(int i=1;i<=n;i++)
  {
      for(int j=1;j<=m;j++)
      {
          insert(i,j,i,j,a[i][j]);    //构建差分数组
      }
  }

题目:

输入一个 nm 列的整数矩阵,再输入 q 个操作,每个操作包含五个整数 x1,y1,x2,y2,其中 (x1,y1)(x2,y2)表示一个子矩阵的左上角坐标和右下角坐标。

每个操作都要将选中的子矩阵中的每个元素的值加上 c

请你将进行完所有操作后的矩阵输出。

输入格式

第一行包含整数 n,m,q

接下来 n 行,每行包含 m个整数,表示整数矩阵。

接下来 q 行,每行包含 5 个整数 x1,y1,x2,y2表示一个操作。

输出格式

n 行,每行 m 个整数,表示所有操作进行完毕后的最终矩阵。

数据范围

1≤n,m≤1000

1≤q≤100000

1≤x1≤x2≤n

1≤y1≤y2≤m

−1000≤c≤1000

−1000≤矩阵内元素的值≤1000

输入样例:
3 4 3
1 2 2 1
3 2 2 1
1 1 1 1
1 1 2 2 1
1 3 2 3 2
3 1 3 4 1
输出样例:
2 3 4 1
4 3 4 1
2 2 2 2

代码如下:

#include<iostream>
#include<dstdio>
using namespace std;
const int N = 100010;
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()
{
    int n, m, q;
    cin >> n >> m >> q;
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= m; j++)
            cin >> 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, 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++)
        {
            b[i][j] += b[i - 1][j] + b[i][j - 1] - b[i - 1][j - 1];  //二维前缀和
        }
    }
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= m; j++)
        {
            printf("%d ", b[i][j]);
        }
        printf("\n");
    }
    return 0;
}

总的来说一维差分的核心是:第一步,创建差分数组b[i]=a[i]-a[i-1],第二步,b[l] + = c, b[r+1] - = c,第三步,a[i]=b[i]+a[i-1].

二维差分的核心就是上面的insert函数。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值