一维前缀和&一维差分(下篇讲解二维前缀和&二维差分)(超详细,python版,其他语言也很轻松能看懂)

本篇博客讲解一维前缀和,一维差分,还会给出一维差分的模板题,下篇博客讲解 二维前缀和&二维差分

一维前缀和:

接触过算法的小伙伴应该都了解前缀和,前缀和在算法中应用很广,不了解也没有关系,这里简单介绍一下什么是前缀和。

如数组a[0,1,2,3,4],则数组a的前缀和为b[0,1,3,6,10],也就是b[i] = a[i] + b[i - 1],利用前缀和我们可以快速求出在数组a在区间[l,r]中的和为多少,如在[2,3]区间内数组a的和为b[3]-b[1] = 5。一维前缀和的递推公式为b[i] = a[i] + b[i - 1]

一维差分:

类似于数学中的求导和积分,差分可以看成前缀和的逆运算。

什么意思?简单来说,如果存在一个数组a,数组a的前缀和数组为b,则数组a就是数组b的差分数组!b是a的前缀和数组!

简单说一种情况,有一个数组,现在需要对数组进行多次操作,每次操作都是在区间[l,r]上加上一个相同的数或者减掉一个相同的数,所有操作完成后,数组为多少?

如果暴力法的话,每次操作的时间复杂度都为O(n),如果操作次数一多,算法可能就会超时,这是就需要用差分来处理这类问题,差分会将每次操作的时间复杂度都降为O(1)。

差分数组:

首先给定一个原数组a:a[1], a[2], a[3] , , , , a[n];

然后我们构造一个数组b : b[1] ,b[2] , b[3] , , , b[i];

使得 a[i] = b[1] + b[2 ]+ b[3] +, , , , + b[i]

也就是说,a数组是b数组的前缀和数组,反过来我们把b数组叫做a数组的差分数组。换句话说,每一个a[i]都是b数组中从头开始的一段区间和。

考虑如何构造差分b数组?

最为直接的方法

如下:

a[0 ]= 0;

b[1] = a[1] - a[0];

b[2] = a[2] - a[1];

b[3] =a [3] - a[2];

b[n] = a[n] - a[n-1];

差分时始终要记得,a数组是b数组的前缀和数组,比如对b数组的b[i]的修改,会影响到a数组中从a[i]及往后的每一个数。

首先让差分b数组中的 b[l] + c ,a数组变成 a[l] + c ,a[l+1] + c, a[n] + c;

然后我们打个补丁,b[r+1] - c, a数组变成 a[r+1] - c,a[r+2] - c,a[n] - c;

为啥还要打个补丁?

我们画个图理解一下这个公式的由来:
在这里插入图片描述
b[l] + c,效果使得a数组中 a[l]及以后的数都加上了c(红色部分),但我们只要求l到r区间加上c, 因此还需要执行 b[r+1] - c,让a数组中a[r+1]及往后的区间再减去c(绿色部分),这样对于a[r] 以后区间的数相当于没有发生改变。

因此我们得出一维差分结论:给a数组中的[ l, r]区间中的每一个数都加上c,只需对差分数组b做 b[l] + = c, b[r+1] - = c。时间复杂度为O(1), 大大提高了效率。

差分数组加减完后,由公式b[i] = a[i] - a[i - 1] 逆推出 前缀和数组a[i] = b[i] + a[i - 1]

题目:差分

题目链接差分

输入一个长度为 n的整数序列。接下来输入 m个操作,每个操作包含三个整数 l,r,c,表示将序列中 [l,r]之间的每个数加上 c。请你输出进行完所有操作后的序列。

输入格式
第一行包含两个整数 n和 m。
第二行包含 n个整数,表示整数序列。
接下来 m行,每行包含三个整数 l,r,c,表示一个操作。

输出格式
共一行,包含 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

代码及详细注释:

n, m = map(int, input().split())  # 输入两个整数n和m
a = list(map(int, input().split()))  # 输入n个整数,存储在列表a中
a = [0, *a]  # 在列表a的开头插入一个0
b = [0] * (n + 2)  # 初始化长度为n+2的全0列表b

# 计算列表b中每个元素的值
for i in range(1, n + 1):
    b[i] = a[i] - a[i - 1]

# 根据输入的操作更新列表b的值
for _ in range(m):
    l, r, c = map(int, input().split())
    b[l] += c
    if r != n: # 判断是否会出界,如果b数组定义的很长,则无需判断
        b[r + 1] -= c
        
# 也可以这样写,但空间复杂度会高一点
# res = [0] * n
# res[0] = b[1]
# for i in range(2, n + 1):
#     res[i - 1] = b[i] + res[i - 2]
# print(" ".join(map(str, res)))

# 计算最终结果
for i in range(1, n + 1):
    b[i] += b[i - 1]

# 输出最终结果,去除首尾两个元素并将列表转换为字符串输出
print(' '.join(map(str, b[1:-1])))


总结:

前缀和&差分可以说是算法竞赛中必考的知识点,需要熟练掌握。
一维差分总结就两个公式:

  1. b[l] + = c, b[r+1] - = c
  2. b[i] = a[i] - a[i - 1]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

不染_是非

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值