python差分法

差分法

在做LeetCode第1109题航班预订统计时，因为时间复杂度太高没有通过，看了官方的解答后，学到了一种新的算法——差分法。

暴力求和和差分法的比较

我们在对一个列表逐项叠加一个相同的数时，我们要遍历整个列表，所以时间复杂度是o(n),但是，我们对数据进行批量处理的时候，总会遇到多个数，要我们逐项叠加，且叠加位置不一样的时候。这里举个例子：

比如我们有两个数组: a = [[1,3,10],[1,4,5]],b = [0]*5这两个数组，(a中1,3表示的是位置，10表示+10)
我们的目标是将a数组表示的信息加到b数组里
0   0   0   0   0  
10  10  10
5   5   5   5
15  15  15  5
结果是c = [15,15,15,5,0]

好，现在我们如果用代码写一下，最简单的方法是双循环：

a = [[1,3,10],[1,4,5]]
b = [0]*5
for i in a:
    x1 = i[0]
    x2 = i[1]
    x3 = i[2]
    for j in range(x1-1,x2):
        b[j] += x3

这样我们就把b变成了[15,15,15,5,0] （c)
实现了加和。
这种方法是很简单的，时间复杂度为o(n的二次方)

差分法：

a = [[1,3,10],[1,4,5]]
b = [0]*5
for left,right,inc in a:
	b[left-1] += a[inc]
	if right < len(a):
		b[right] -= a[inc]
for i in range(1,n):
	n[i] += n[i-1]
print(n)

时间复杂度为o(n+m)

学习差分法

下面是leetcode，对于差分法的讲解：

差分数组对应的概念是前缀和数组，对于数组 [1,2,2,4]，其差分数组为 [1,1,0,2]，差分数组的第 ii 个数即为原数组的第 i-1个元素和第 i个元素的差值，也就是说我们对差分数组求前缀和即可得到原数组。

差分数组的性质是，当我们希望对原数组的某一个区间 [l,r] 施加一个增量inc 时，差分数组 d对应的改变是：d[l] 增加 inc，d[r+1] 减少 inc。这样对于区间的修改就变为了对于两个位置的修改。并且这种修改是可以叠加的，即当我们多次对原数组的不同区间施加不同的增量，我们只要按规则修改差分数组即可。

就比如：一个数组[0,0,0,0,0,0]在[1,4]每个数都加10，那么差分数组就是[10,0,0,0,0,-10]，2和5中间有四个数，所以是四个0，每个数都跟前一个数做比较，取差值。
取完差值后再叠加，最后让每一项逐一加上前一项。

最后来看道例题

出自leetcode1109.航班预订统计
这里有 n 个航班，它们分别从 1 到 n 进行编号。

有一份航班预订表 bookings ，表中第 i 条预订记录 bookings[i] = [firsti, lasti, seatsi] 意味着在从 firsti 到 lasti （包含 firsti 和 lasti ）的 每个航班 上预订了 seatsi 个座位。

请你返回一个长度为 n 的数组 answer，其中 answer[i] 是航班 i 上预订的座位总数。
示例 1：

输入：bookings = [[1,2,10],[2,3,20],[2,5,25]], n = 5
输出：[10,55,45,25,25]
解释：
航班编号 1 2 3 4 5
预订记录 1 ： 10 10
预订记录 2 ： 20 20
预订记录 3 ： 25 25 25 25
总座位数： 10 55 45 25 25
因此，answer = [10,55,45,25,25]

解答：

class Solution:
    def corpFlightBookings(self, bookings, n: int):
        num = [0]*n
        for left, right, inc in bookings:
            num[left-1] += inc
            if right < n:
                num[right] -= inc
        for i in range(1,n):
            num[i] += num[i-1]
        return num