差分数组及使用

什么是差分数组以及应用

B站讲解差分数组及应用
最常用的就是修改数组中的数，一段一段的修改，每一次的修改都是独立的
比如：
list = [1, 2, 3, 4, 5, 6]
修改一： 0-2 统统加1
修改二： 1-3 统统减1
问这个数组变成啥样了？

什么是差分数组？
原数组 [1, 2, 3, 4, 5, 6]
它的差分数组就是 除第一个不变外 后一个减去前一个形成的新的数组
[1, 1, 1, 1, 1, 1]

什么是前缀和？
先要有个差分数组，后一个加前一个得到的数组就是前缀和
[1, 1, 1, 1, 1, 1] ->>>>>>>> [1, 2, 3, 4, 5, 6]

根据前缀和可以还原出原数组

如何使用？

对于一个待修改的数组，先求他的差分数组，次数以 [1, 2, 3, 4, 5, 6] 为例！
它的差分数组为：
[1, 1, 1, 1, 1, 1]

修改一： 0-2 统统加1 .那么差分数组的第0个加1(注意只有这个加1)， 第2个的后一个减1（注意是结束的那个的后一个！如果越界了就不减！）
[1, 1, 1, 1, 1, 1] ->>>>>>>> [2, 1, 1, 0, 1, 1]

修改二： 1-3 统统减1 那么差分数组的第1个减1， 第4个加1

[2, 1, 1, 0, 1, 1] -->>>>>> [2, 0, 1, 0, 2, 1]

将最后得到的差分数组求前缀和即可：
[2, 0, 1, 0, 2, 1] ->>>>>>> [2, 2, 3, 3, 5, 6]

典型的题目：
leetcode M1109

def M1109(bookings: list[list[int]], n: int) -> list[int]:
    # 构建差分数组，这里开始全是0
    diff = [0 for _ in range(n)]
    for i in bookings:
        diff[i[0]-1] += i[2]
        # 注意判断是否到达边界
        # 注意i[1],i[0] 不是索引，它们减1才是索引！
        if i[1] <= n-1:
            diff[i[1]] -= i[2]
    for i in range(1, n):
        diff[i] = diff[i-1]+diff[i]
    return diff

二维差分

差分不仅可以用在以为的数组上实现高效的数据更改，也可以使用在矩阵上实现二维数据的高效更改

二维差分也很简单，只需要记住，当差分数组diff的第i行第j列+1后，对应原来的数组中从 [ i ] [ j ]这个点一直到整个矩阵的右下角那个点这个一大块都会+1，所以，如果只需要对(x1,y1),(x2,y2)这个范围内的矩形进行+d操作的话要进行4步操作:

diff[x1][y1]+=d
diff[x1][y2+1] -= d  # 当然要格外注意数组越界的问题
diff[x2+1][y1] -= d
diff[x2+1][y2+1] += d

M6292 子矩阵元素+1
解答如下：

def rangeAddQueries(n: int, queries: List[List[int]]) -> List[List[int]]:
    # 这里这个ans就是差分数组
    ans = [[0 for _ in range(n)] for _ in range(n)]

    for r1, c1, r2, c2 in queries:
        ans[r1][c1] += 1
        # 下面的3个if是用来判断是否越界的
        if c2 + 1 < n:
            ans[r1][c2 + 1] -= 1
        if r2 + 1 < n:
            ans[r2 + 1][c1] -= 1
        if r2 + 1 < n and c2 + 1 < n:
            ans[r2 + 1][c2 + 1] += 1
    # 将差分数组转换成阵阵的数组之前先要将第一行和第一列转换下，可以节省很多事
    for i in range(1, n):
        ans[0][i] = ans[0][i] + ans[0][i - 1]
        ans[i][0] = ans[i][0] + ans[i - 1][0]
    # 将差分数组转换成真正的数组
    for i in range(1, n):
        for j in range(1, n):
            # 当前这个位置对应的真正的数 = 前一个+上一个-左上角那个+原来差分这个位置对应的一个
            ans[i][j] = ans[i - 1][j] + ans[i][j - 1] - ans[i - 1][j - 1] + ans[i][j]
    return ans

但是如何将原来的二维数组转换成差分数组呢？

a = [[1, 2, 3] for _ in range(3)]

# 首先构造一个和原来数组一样大小的差分数组，初始化全是0
diff = [[0] * 3 for _ in range(3)]
for i in range(3):
    for j in range(3):
        if i == 0 and j == 0:
            diff[i][j] = a[i][j]
            continue
        # i==0,j==0 这两个直接降维成了一维数组差分
        if i == 0:
            diff[i][j] = a[i][j] - a[i][j - 1]
            continue
        if j == 0:
            diff[i][j] = a[i][j] - a[i - 1][j]
            continue
        # 下面这行才是关键
        diff[i][j] = a[i][j] - a[i - 1][j] - a[i][j - 1] + a[i - 1][j - 1]